Darth Vader and Tree【DP+矩阵快速幂】

题目链接

---

  我们再不考虑x大小的时候，我们可以直接：

  其中，len(j)是j这个长度的出现次数，也就是输入的。

  但是，我们知道，上式的i可能会很大，这里就得要想办法优化，由于这里的d≤100，我们可以在这里考虑优化。

  用矩阵维护吧，因为这里有矩阵的性质。

那么，我们就可以去构造矩阵了，矩阵的大小是101 * 101。因为我们的长度是1～100，所以其中可能会用到dp(1)~dp(100)的值（其实只用到99就可以了，但是方便嘛）。然后就是构造矩阵了。

  写在代码里了。
 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 1e5 + 7;
const ll mod = 1e9 + 7;
int N, X, d[maxN], len[maxN] = {0};
ll dp[maxN] = {0}, sum[maxN] = {0};
struct matrice
{
    ll a[105][105];
    matrice() { memset(a, 0, sizeof(a)); }
    friend matrice operator * (matrice x, matrice y)       //重载函数，矩阵的乘积得到的新矩阵
    {
        matrice ans;
        for(int i=0; i<=100; i++)
        {
            for(int j=0; j<=100; j++)
            {
                ll tmp = 0;
                for(int k=0; k<=100; k++)
                {
                    tmp=(tmp + x.a[i][k] * y.a[k][j]) % mod;       //最后得到的需要取mod
                }
                ans.a[i][j] = tmp;
            }
        }
        return ans;
    }
    void OUT() { for(int i=0; i<=100; i++) for(int j=0; j<=100; j++) printf("%lld ", a[i][j]); puts(""); }
}Bas, AI, Want;
matrice fast_mi(matrice x, ll ti)
{
    matrice ans;
    for(int i=0; i<=100; i++) ans.a[i][i] = 1;  //构造单位阵
    while(ti)
    {
        if(ti & 1) ans = ans * x;
        x = x * x;
        ti >>= 1;
    }
    return ans;
}
int main()
{
    scanf("%d%d", &N, &X);
    for(int i=1; i<=N; i++) { scanf("%d", &d[i]); len[d[i]]++; }
    dp[0] = 1; sum[0] = 1;
    for(int i=1; i<=100; i++)
    {
        for(int j=i; j>0; j--)
        {
            dp[i] = (dp[i] + dp[i - j] * len[j]) % mod;
        }
        sum[i] = (sum[i - 1] + dp[i]) % mod;
    }
    if(X <= 100) { printf("%lld\n", sum[X]); return 0; }    //不需要矩阵优化
    for(int i=0; i<99; i++)
    {
        Bas.a[i][i + 1] = 1;
    }
    Bas.a[100][100] = 1;  //s(n) = s(n - 1) + dp(n)---(s(n - 1))
    for(int i=0; i<100; i++)    //dp(n) = segma(dp(n - k) * len(k))
    {
        Bas.a[99][i] = len[100 - i];    //dp(n)---solve
        Bas.a[100][i] = len[100 - i];   //s(n) = s(n - 1) + dp(n)---(dp(n))
    }
//    Bas.OUT();
    Bas = fast_mi(Bas, X - 100);
    for(int i=0; i<100; i++) AI.a[i][0] = dp[i + 1];
    AI.a[100][0] = sum[100];
    Want = Bas * AI;
    printf("%lld\n", Want.a[100][0]);
    return 0;
}