随机化算法比较矩阵相等 Matrix God

I - Matrix God Gym - 101341I

这道题很直白,给出矩阵A,B,C;要你计算A*B是否等于C;
普通的O(n3)方法一定超时,我一直在尝试用Strassen递归的方法,发现随机化算法会简单很多,随机构造两个矩阵r1和r2,一个是1*n维,一个是n*1维,将r1*a*b*r2计算出来,就是一个整数,然后在计算r1*c*r2,结果也是一个整数。比对两个整数是否相等,这是答案的必要不充分条件,所以我们多尝试几组这样的r1,r2就可以不断的逼近真实。

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int maxn=3000+5;
const int mod = 1e9+7;
LL a[maxn][maxn],b[maxn][maxn],c[maxn][maxn];
LL res1[maxn],res2[maxn],r1[maxn],r2[maxn],resL,resR;

int main()
{
    srand(unsigned(time(NULL)));
    //freopen("input.in","r",stdin);
    int n;
    while(scanf("%d",&n)==1) {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        memset(c,0,sizeof(c));
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%lld",&a[i][j]);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%lld",&b[i][j]);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%lld",&c[i][j]);

        bool door=true;
        for(int z =0;z<5;z++)
        {
            memset(res1,0,sizeof(res1));
            memset(res2,0,sizeof(res2));
            for(int i=0;i<n;i++)
            {
                r1[i] = rand();
                r2[i] = rand();
            }
            for(int i=0;i<n;i++)
                for(int j=0;j<n;j++)
                {
                    res1[i]+=(r1[j]*a[j][i])%mod;
                    res2[i]+=(b[i][j]*r2[j])%mod;
                    res1[i]%=mod;
                    res2[i]%=mod;
                }

            resL=resR=0;
            for(int i=0;i<n;i++)
                resL=(resL+(res1[i]*res2[i])%mod)%mod;

            memset(res1,0,sizeof(res1));
            for(int i=0;i<n;i++)
                for(int j=0;j<n;j++)
                    res1[i]=(res1[i]+(r1[j]*c[j][i])%mod)%mod;


            for(int i=0;i<n;i++)
                resR=(resR+(res1[i]*r2[i])%mod)%mod;
            if(resL!=resR){
                door=false;
                break;
            }
        }
        if(door)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_37957829/article/details/79980495
个人分类: 随机化算法
上一篇CSU 2070 Seating Chart
下一篇POJ3264 MRQ算法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭