网易2017实习生笔试编程题-魔力手环

意思大概就是循环累加，累加的次数上限有2的9次幂，结果直接超时了

现附上别人的思路

把手环数字转为一个向量,然后乘

     [1 1 0 0 0]
     [0 1 1 0 0]
     [0 0 1 1 0]
     [0 0 0 1 1]
     [1 0 0 0 1]

这个矩阵N次即可。。用个矩阵快速幂,边算边求模。

作者：NotDeep
链接：https://www.nowcoder.com/discuss/22696?type=0&order=0&pos=5&page=1
来源：牛客网

#include <bits/stdc++.h>

using namespace std;

void mult(int A[50][50], int B[50][50], int C[50][50], int n) {
    int T[50][50];
    memset(T, 0, sizeof(T));
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            for(int k = 0; k < n; k++) {
                T[i][j] = (T[i][j] + A[i][k] * B[k][j]) % 100;
            }
        }
    }
    memcpy(C, T, sizeof(T));
}
void mypower(int A[50][50], int R[50][50], int n, int m) {
    if(n == 0) {
        memset(R, 0, sizeof(int) * 2500);
        for(int i = 0; i < m; i++) R[i][i] = 1;
    } else if(n % 2 == 0) {
        mypower(A, R, n/2, m);
        mult(R, R, R, m);
    } else {
        mypower(A, R, n - 1, m);
        mult(A, R, R, m);
    }
}
vector<int> solve(vector<int> seq, int m) {
    int A[50][50], R[50][50], n = (int)seq.size();
    memset(A, 0, sizeof(A));
    for(int i = 0; i < n; i++) A[i][i] = A[i][(i + 1) % n] = 1;
    mypower(A, R, m, n);
    vector<int> res;
    for(int i = 0; i < n; i++) {
        int sum = 0;
        for(int j = 0; j < n; j++) sum = (sum + R[i][j] * seq[j]) % 100;
        res.push_back(sum);
    }
    return res;
}
int main () {
    int n, k;
    vector<int> x;
    cin >> n >> k;
    for(int i = 0; i < n; i++) {
        int tmp; cin >> tmp;
        x.push_back(tmp);
    }
    vector<int> ans = solve(x, k);
    for(int i = 0; i < ans.size(); i++) i == 0 ? cout << ans[i] : cout << " " << ans[i];
    return 0;
}