poj3150 矩阵

最新推荐文章于 2020-12-11 12:06:47 发布

ascii991

最新推荐文章于 2020-12-11 12:06:47 发布

阅读量381

点赞数

分类专栏： acm_数学文章标签： ini 优化 c

本文链接：https://blog.csdn.net/ascii991/article/details/7448134

版权

acm_数学专栏收录该内容

9 篇文章 0 订阅

订阅专栏

（我想写的和这位博主一样，就转载了，表示这是一道好题，对矩阵乘法的优化有所启示）

【题意】

一个n个位置的环，每经过一次置换，每个位置的值会变成上一次周围d范围之内的和，求mod m后置换过k次后每个位置的值

很容易可以看出是矩阵乘法

n<=500

如果构建n*n的矩阵，矩阵乘法复杂度为n^3，那么总体复杂度为n^3log k 是不能接受的

但是可以发现一个矩阵的性质

因为是求d范围之内之和，所以矩阵每一行基本一样，只是错了一位而已

所以求矩阵的时候只需要求一行就够了，复杂度降到n^2，便可以通过了。

（我想写的和这位博主一样，就转载了，表示这是一道好题，对矩阵乘法的优化有所启示）

（代码是自己写的）

#include <iostream>
using namespace std;
const int maxn=505; 
long long a[maxn],ans[maxn],c[maxn];
int n,m,d,k,i,j;
void mul(long long a[],long long b[])
{
     long long c[maxn]={0};
     int i,j;
     for (i=0;i<n;i++)
         for (j=0;j<n;j++)
             c[i]+=a[j]*b[(i>=j)?i-j:i-j+n];
     for (i=0;i<n;i++)
         a[i]=c[i]%m;         
} 
int main()
{
    freopen("pin.txt","r",stdin);
    freopen("pou.txt","w",stdout);
    cin >> n >> m >> d >> k;
    for (i=0;i<n;i++)
        cin >> c[i];
    for (a[0]=i=1;i<=d;i++)
        a[i]=a[n-i]=1;
    ans[0]=1;
    while (k)
    {
          if (k&1) 
             mul(ans,a);
          mul(a,a);
          k>>=1;
    }
    for (i=0;i<n;i++)
    {
        long long ss=0;
        for (j=0;j<n;j++)
            ss=(ss+ans[(j>=i)?j-i:j-i+n]*c[j])%m;    
        cout << ss;
        if (i!=n-1) cout << " ";
        else cout << endl;
    } 
    return 0;
}