Just Pour the Water ZOJ - 2974 矩阵快速幂

关注

原创最新推荐文章于 2026-03-14 11:23:54 发布 373 阅读 AI 写同款· GEO 优化 ›

YOLO26涨点改进 | 独家创新，特征融合改进篇 | TGRS 2025 | 引入FFM特征融合模块，实现特征的全局交互与融合，含多种组合创新改进，适合小目标检测、助力YOLO26有效涨点本文提出了一种改进YOLO26目标检测模型的FFM模块，该模块通过跨模态注意力机制实现全局特征融合，显著提升模型对小目标、密集目标和复杂场景的检测性能。FFM采用多头跨注意力机制建模不同模态间的复杂关系，同时引入可调节下采样率降低计算复杂度。实验表明，在WHU-OPT-SAR等数据集上，FFM模块使mIoU提升0.75%，OA提升0.32%。文章详细介绍了FFM的结构原理、核心代码实现，并提供了两种改进YOLO26的yaml配置文件方案，包括FFM单独使用以及与FCM模块的组合方案。阅读详情

思路：

用一个N*N的矩阵的第一行来记录水量。

另外一个矩阵记录转移的方式，不是很好描述，具体看代码。

当k是0的时候，水量都给了自己。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n;
const int N = 25;
struct Matrix {
    double m[N][N];
    Matrix(){};
    Matrix operator*(Matrix const &a) const {
        Matrix res;
        memset( res.m,0,sizeof res.m ) ;
        for ( int i=0; i<n; i++ ) {
            for ( int j=0; j<n; j++ ) {
                for (int k=0;k<n;k++ ) {
                    res.m[i][j] += this->m[i][k]*a.m[k][j];
                }
            }
        }
        return res;
    }
};
Matrix res;
void pow_mod( Matrix base ,int n ) {

    while ( n>0 ) {
        if ( n&1 ) res = res*base;
        base = base*base;
        n>>=1;
    }
}
int main()
{
    int T;
    cin>>T;
    while ( T-- ) {
        scanf("%d",&n);
        memset( res.m,0,sizeof res.m );
        for ( int i=0; i<n; i++ )
            scanf("%lf",&res.m[0][i]);
        int k;
        Matrix base;
        memset( base.m,0,sizeof base.m );
        for ( int i=0; i<n; i++ ) {
            scanf("%d",&k);
            if ( k==0 ) {
                base.m[i][i] = 1;
                continue;
            }
            for ( int j=0; j<k; j++ ) {
                int num;
                scanf("%d",&num);
                base.m[i][num-1] = 1.0/k;
            }
        }
        int time;
        scanf("%d",&time);
        pow_mod( base,time ) ;
        for ( int i=0; i<n; i++ )
            if ( i==n-1 ) printf("%.2f\n",res.m[0][i]);
            else printf("%.2f ",res.m[0][i]);
    }


    return 0;
}

收录于