HDU 4920 Matrix multiplication

最新推荐文章于 2024-07-31 10:00:00 发布

原创最新推荐文章于 2024-07-31 10:00:00 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#matrix #bitset

本文探讨了在处理模数为3的问题时，如何利用bitset进行优化，通过实例展示了一种高效的解决方法。

因为模数为3，所以可以用bitset优化。。。简直666

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
#include <bitset>

using namespace std;

int n,m,a,b,x,y;

bitset <1005> x_1[805],x_2[805],y_1[805],y_2[805];

int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d",&n)==1)
    {
        for(int i=0;i<=n;++i)
        {
            x_1[i].reset();x_2[i].reset();
            y_1[i].reset();y_2[i].reset();
        }
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<n;++j)
            {
                scanf("%d",&a);
                if(a%3==1)x_1[i].set(j);
                else if(a%3==2)x_2[i].set(j);
            }
        }
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<n;++j)
            {
                scanf("%d",&a);
                if(a%3==1)y_1[j].set(i);
                else if(a%3==2)y_2[j].set(i);
            }
        }
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<n;++j)
            {
                x=(x_1[i]&y_1[j]).count()+(x_2[i]&y_2[j]).count();
                y=(x_1[i]&y_2[j]).count()+(x_2[i]&y_1[j]).count();
                b=x+y*2;b%=3;
                if(j)putchar(' ');
                printf("%d",b);
            }
            puts("");
        }
    }
    return 0;
}