24h購物| | PChome| 登入
2012-11-27 17:03:37| 人氣339| 回應0 | 上一篇 | 下一篇

[PTC][201211] PA Circular Matrix Product

推薦 0 收藏 0 轉貼0 訂閱站台

這題不算難,難是難在卡時間,很清楚地我們知道要進行矩陣連乘積,
也就是 N*N 的矩陣連乘 E-S 次,最後乘上一個 N*1 的矩陣。
不能直接做矩陣乘法,因為會消耗 O(N*N*N) 計算兩個 N*N 相乘。

在此倒過來運算 (N*N)*(N*1) = (N*1) 消耗 O(N*N),
逆序運算得 O(N*N*(S-E))。
直接算 O(N*N*N*(S-E))。

或許有人會想用線段樹降低 O(E-S),在此先說不僅僅是 MLE 而且還會 TLE。



#include <stdio.h>
#include <string.h>
struct mx {
    unsigned char v[55][55];
};
mx mult(mx A, mx B, int a, int b, int c){
    mx C;
    int i, j, k;
    memset(&C, 0, sizeof(C));
    for(i = 0; i < a; i++) {
        for(j = 0; j < c; j++) {
            for(k = 0; k < b; k++)
                C.v[i][j] += A.v[i][k]*B.v[k][j];
        }
    }
    return C;
}
mx A[855], B, one;
int M, N, K, S, E;
/*
3 2 1
10 100 200 255
8 6 2 3
1 0 0 1
1 0 2 8
*/
int main() {
    int i, j, k, x;
    memset(&one, 0, sizeof(one));
    for(i = 0; i < 55; i++)
        one.v[i][i] = 1;
    while(scanf("%d %d %d", &M, &N, &K) == 3) {
        for(i = 0; i < M; i++) {
            for(j = 0; j < N; j++)
                for(k = 0; k < N; k++) {
                    scanf("%d", &x);
                    A[i].v[j][k] = (unsigned char)x;
                }
        }
        mx ans;
        while(K--) {
            scanf("%d %d", &S, &E);
            for(i = 0; i < N; i++) {
                scanf("%d", &x);
                B.v[i][0] = x;
            }
            ans = B;
            if(S <= E) {
                for(i = E; i >= S; i--)
                ans = mult(A[i], ans, N, N, 1);
            } else {
                for(i = E; i >= 0; i--)
                ans = mult(A[i], ans, N, N, 1);
                for(i = M-1; i >= S; i--)
                ans = mult(A[i], ans, N, N, 1);
            }
            for(i = 0; i < N; i++) {
                if(i)
                    printf(" ");
                printf("%d", ans.v[i][0]);
            }
            puts("");
        }
    }
    return 0;
}

台長: Morris
人氣(339) | 回應(0)| 推薦 (0)| 收藏 (0)| 轉寄
全站分類: 不分類 | 個人分類: 資訊競賽 |
此分類下一篇:[PTC][201211] PB Job Allocation
此分類上一篇:[ZJ][三年後修訂] d371. 3. Huffman 編碼中的編碼效能問題

是 (若未登入"個人新聞台帳號"則看不到回覆唷!)
* 請輸入識別碼:
請輸入圖片中算式的結果(可能為0) 
(有*為必填)
TOP
詳全文