[网易2017实习生编程题] 魔力手环

【Problem 5】分饼干

题目

小易拥有一个拥有魔力的手环上面有n个数字(构成一个环),当这个魔力手环每次使用魔力的时候就会发生一种奇特的变化：每个数字会变成自己跟后面一个数字的和(最后一个数字的后面一个数字是第一个),一旦某个位置的数字大于等于100就马上对100取模(比如某个位置变为103,就会自动变为3).现在给出这个魔力手环的构成，请你计算出使用k次魔力之后魔力手环的状态。

输入描述:

输入数据包括两行：
第一行为两个整数n(2 ≤ n ≤ 50)和k(1 ≤ k ≤ 2000000000),以空格分隔
第二行为魔力手环初始的n个数，以空格分隔。范围都在0至99.

输出描述:

输出魔力手环使用k次之后的状态，以空格分隔，行末无空格。

输入例子:

3 2
1 2 3

输出例子:

8 9 7

解答

方法一

一开始直接按照题意模拟编程，但是由于k值过大，从而超时，牛客网上的通过率只有百分之六十。

#include <vector>
#include <iostream>

using namespace std;

int main(int argc, char *argv[])
{
    int n,k; 
    cin >> n;
    cin >> k;
    vector<int> data(n);
    for (int i = 0; i < n; ++i)
    {
        cin >> data[i];
    }

    for (int i = 0; i < k; ++i)
    {
        int temp = data[0], n = data.size();
        for (int i = 0; i < n - 1; ++i)
        {
            data[i] = (data[i] + data[i + 1]) % 100;
        }
        data[n - 1] = (data[n - 1] + temp) % 100;
    }

    for (int i = 0; i < n; ++i)
    {
        (i == 0) ? cout << data[i] : cout << " " << data[i];
    }
    cout << endl;
    return 0;
}

方法二

后来参考了网上的解法，利用矩阵快速幂算法可以降低算法的复杂度。

将手环上的数字看做是一个只有一列的矩阵，如果一个手环上有5个数字，如果计算其变化k次后的结果，只需要左乘如下矩阵k次就可以了。

[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]

我们可以考虑使用矩阵快速幂算法,先计算如上矩阵进行k次幂运算后的结果，然后在和手环上的数字向量相乘。

# include <iostream>
# include <vector>
# include <algorithm>

using namespace std;

vector<vector<int>> multiMatrix(vector<vector<int>>& A, vector<vector<int>>& B)
{
    int n = A.size();
    vector<vector<int>> res(n, vector<int>(n));
    for (int i = 0; i < n; ++i)
    {
        for (int j = 0; j < n; ++j)
        {
            for (int k = 0; k < n; ++k)
            {
                res[i][j] = (res[i][j] + A[i][k] * B[k][j ]) % 100;
            }
        }
    }
    return res;
}

vector<vector<int>> getMatrixPower(vector<vector<int>>& A, int k)
{
    int n = A.size();
    vector<vector<int>> temp(n,vector<int>(n));
    for (int i = 0; i < n; ++i) { temp[i][i] = 1; }
    while (k)
    {
        if (k % 2)
        {
            temp = multiMatrix(temp, A);
        }
        k = k >> 1;
        A = multiMatrix(A, A);
    }
    return temp;
}

vector<int> solve(vector<int>& data, int k)
{
    int n = data.size();
    vector<vector<int>> base(n, vector<int>(n));
    for (int i = 0; i < n; ++i)
    {   
        base[i][i] = base[i][(i + 1) % n] = 1;
    }
    vector<vector<int>> targ = getMatrixPower(base, k);

    vector<int> res(n);
    for (int i = 0; i < n; ++i)
    {
        int sum = 0;
        for (int j = 0; j < n; ++j)
        {
            sum = (sum + targ[i][j] * data[j]) % 100;
        }
        res[i] = sum;
    }
    return res;
}

int main(int argc, char *argv[])
{
    int n, k;
    cin >> n, cin >> k;
    vector<int> data(n);
    for (int i = 0; i < n; ++i) { cin >> data[i]; }

    data = solve(data, k);
    for (int i = 0; i < n; ++i)
    {
        (i == 0) ? cout << data[i] : cout << " " << data[i];
    }
    cout << endl;
}