程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> hdu 4920 Matrix multiplication(高效)

hdu 4920 Matrix multiplication(高效)

編輯:C++入門知識

hdu 4920 Matrix multiplication(高效)


題目鏈接:4920 Matrix multiplication

題目大意:給定兩個n階矩陣,求矩陣相乘後模3.

解題思路:因為矩陣模掉3後只有0,1,2三種情況。所以對於矩陣A,記錄每一行中1,2的位置,借助bitset。矩陣B中每一列1,2的位置。然後對於結果中每個位置,只要考慮1?1,1?2,2?1,2?2的個數即可。

#include 
#include 
#include 
#include 

using namespace std;
const int maxn = 805;

int N, C[maxn][maxn];
bitset x[maxn][2], y[maxn][2];

void init () {
    int u;
    memset(C, 0, sizeof(C));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < 2; j++) {
            x[i][j].reset();
            y[i][j].reset();
        }
    }

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            scanf("%d", &u);
            u %= 3;
            if (u)
                x[i][u-1].set(j, 1);
        }
    }

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            scanf("%d", &u);
            u %= 3;
            if (u)
                y[j][u-1].set(i, 1);
        }
    }
}

int solve (int u, int v) {
    int ret = 0;
    for (int i = 0; i < 2; i++) {
        for (int j = 0; j < 2; j++) {
            bitset k = x[u][i]&y[v][j];

            ret = (ret + (i+1)*(j+1)*k.count()) % 3;
        }
    }
    return ret;
}

int main () {
    while (scanf("%d", &N) == 1) {
        init();
        for (int i = 0; i < N; i++) {
            printf("%d", solve(i, 0));
            for (int j = 1; j < N; j++)
                printf(" %d", solve(i, j));
            printf("\n");
        }
    }
    return 0;
}

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved