HDU - 5171 GTY's birthday gift （矩阵快速幂）

题意：

每次将序列中两个数相加再加入序列中，反复操作k次，问k次操作后的序列和最大是多少。

思路：

序列和最大，显然是每次取序列中最大的两个数相加。但是k最大为10亿，实在太大了，所以需要找规律。

输入

3 2
6 3 2

k=1：6 + 3 = 9
1 * 6 + 1 * 3 = 9

k=2：6 + 3 + 6 = 15
2 * 6 + 1 * 3 = 15

k=3：6 + 3 + 6 + 6 +3 + 6 = 24
3 * 6 + 2 * 3 = 24

k=4：6 + 3 + 6 + 6 +3 + 6 + 6 + 3 + 6 = 39
5 * 6 + 3 * 3 = 39
……..
发现每次6和3前面的系数序列是斐波那契序列，加上k很大，想到利用矩阵快速幂解决。

构造矩阵：

证明一下斐波那契序列 S(n) = F(n+2) - 1
a[8] = a[7] + a[6];
a[7] = a[6] + a[5];
a[6] = a[5] + a[4];
a[5] = a[4] + a[3];
a[4] = a[3] + a[2];
a[3] = a[2] + a[1];
a[2] = a[1] + a[0];

这时候 我们给左右两边同时补个 a[0], a[7];

a[7] = a[7];
a[8] = a[7] + a[6];
a[7] = a[6] + a[5];
a[6] = a[5] + a[4];
a[5] = a[4] + a[3];
a[4] = a[3] + a[2];
a[3] = a[2] + a[1];
a[2] = a[1] + a[0];
a[0] = a[0];

把这 8 个式子相加;

右边 = 2 * S[7];

左边 = S[7] + a[7] + a[8] - a[1] —> S[7] + a[9] - a[1];

两边同时约掉个 S[7], 就变成了 a[9] - a[1] = S[7];

又因为 a[1] = 1, 则 S[n] = a[n+2] - 1;

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
using namespace std;
typedef __int64 ll;
const int INF = 0x3f3f3f3f;
const int MOD = 10000007;
const int SIZE = 2;
struct Matrix {
    ll v[SIZE][SIZE];
    Matrix() {
        memset(v, 0, sizeof(v));
    }
    void init(ll _v) {
        for(int i = 0; i < SIZE; i++)
            v[i][i] = _v;
    }
};
Matrix operator * (Matrix a, Matrix b) {
    Matrix c;
    for(int i = 0; i < SIZE; i++) {
        for(int j = 0; j < SIZE; j++) {
            c.v[i][j] = 0;
            for(int k = 0; k < SIZE; k++) {
                c.v[i][j] += (a.v[i][k] * b.v[k][j]) % MOD;
                c.v[i][j] %= MOD;
            }
        }
    }
    return c;
}
Matrix operator ^ (Matrix a, ll k) {
    Matrix c;
    c.init(1);
    while(k) {
        if(k & 1)
            c = a * c;
        a = a * a;
        k >>= 1;
    }
    return c;
}

const int N = 100005;
int a[N];
bool cmp(int a, int b) {
    return a > b;
}
int main() {
    ll sum, sn, k;
    int n;
    while(scanf("%d %I64d",&n,&k) != EOF) {
        Matrix ret, t, f;
        sum = 0;
        for(int i = 0; i < n; i++) {
            scanf("%d", &a[i]);
            sum += a[i];
            sum %= MOD;
        }
        sort(a, a+n, cmp);
        t.v[0][0] = 1, t.v[0][1] = 1;
        t.v[1][0] = 1, t.v[1][1] = 0;

        //f2            //f1
        f.v[0][0] = 1, f.v[0][1] = 1; 
        f.v[1][0] = 0, f.v[1][1] = 0;


        ret = f * (t^(k+1));
        ll f2 = ret.v[0][0] - 1 - 1;
        ll f1 = ret.v[0][1] - 1;
        sn = (f2 * a[0] % MOD + f1 * a[1] % MOD + sum) % MOD;
        printf("%I64d\n", sn % MOD);
    }
    return 0;
}