bzoj4870

http://www.lydsy.com/JudgeOnline/problem.php?id=4870

矩阵快速幂。。。

人话题意：从nk个物品里选模k余r个物品，问方案数模P

那么我们有方程 f[i][j]=f[i-1][j]+f[i-1][j-1] 跟组合数一个样子 j∈(0,k) 这个物品选还是不选加起来

构造矩阵：x.a[0][0]=1 0个里选0个的方案是1 g.a[i][i]=1 g.a[i][i+1]=1 自己手画一下

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 60;
struct mat {ll a[N][N];
} x, g;
ll n, k, p, r;
void build()
{x.a[0][0] = 1; //0个物品选0个的方案数为1 for(int i = 0; i < k; ++i) {++g.a[i][i]; ++g.a[i][(i + 1) % k]; //矩阵的系数 f[i][j]=f[i-1][j]+f[i-1][j-1] 所以j=1,j-1=1 
    }
}
mat operator * (const mat &A, const mat &B)
{mat ret; memset(ret.a, 0, sizeof(ret.a));for(int i = 0; i < k; ++i)for(int j = 0; j < k; ++j)for(int x = 0; x < k; ++x) ret.a[i][j] = (ret.a[i][j] + A.a[i][x] * B.a[x][j]) % p;return ret;
}
void power(mat A, ll t)
{for(; t; A = A * A, t >>= 1) if(t & 1) x = x * A;
}
int main()
{scanf("%lld%lld%lld%lld", &n, &p, &k, &r);build();power(g, n * k);printf("%lld\n", x.a[0][r]);return 0;
}