bzoj4547

矩阵乘法

看成了合并果子。。。

就是斐波那契数列，只是有负数的时候，先把负数变成正的，然后矩乘

矩乘还是用单位举矩阵记录快速幂的矩阵比较保险

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = ;
const ll mod = 10000007ll;
int n, k;
ll ans;
ll a[N];
struct mat {
    ll a[][];
    mat() { memset(a, , sizeof(a)); }
    mat friend operator * (const mat &a, const mat &b) {
        mat ret;
        memset(ret.a, , sizeof(ret.a));
        for(int i = ; i < ; ++i)
            for(int j = ; j < ; ++j)
                for(int k = ; k < ; ++k) ret.a[i][j] = (ret.a[i][j] + a.a[i][k] * b.a[k][j] % mod) % mod;
        return ret;
    }
} A, B, tmp;
int main()
{
    scanf("%d%d", &n, &k);
    for(int i = ; i <= n; ++i) scanf("%lld", &a[i]), ans = ((ans + a[i]) % mod + mod) % mod;
    sort(a + , a + n + );
    while(a[n - ] <  && k)
    {
        a[n - ] += a[n];
        --k;
        ans = ((ans + a[n - ]) % mod + mod) % mod;
    }
    A.a[][] = ; A.a[][] = ; A.a[][] = ;
    A.a[][] = ; A.a[][] = ; A.a[][] = ;
    A.a[][] = ; A.a[][] = ; A.a[][] = ;
    B.a[][] = a[n];
    B.a[][] = a[n - ];
    B.a[][] = ans;
    for(int i = ; i < ; ++i) tmp.a[i][i] = ;
    for(; k; k >>= , A = A * A) if(k & ) tmp = tmp * A;
    B = tmp * B;
    printf("%lld\n", (B.a[][] % mod + mod) % mod);
    return ;
}