[HDU] A Very Simple Problem

我是链接

这道题做着的思路还是很不容易的，不看题解不会真的去写二项式展开。
首先求出0~50以内的C(n , m)，然后我们又根据二项式定理，知道
$$ (x + 1) ^2 = \sum_{i =0}^n C(n , i) * x^i $$
(个人习惯写法的二项式定理qwq)
所以要推出 $T_n = x^{k + 1} * {k + 1} ^ x $ 需要知道前 n-1 项的 $ T_i $ 所以我们会想到用矩阵快速幂来递推所有 \(T_n\) 。
转换矩阵有很多种。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;
typedef long long ll;
const ll N = 55;

ll Mod,C[N][N];

struct matrix{
	ll M[N][N];
	int l,r;
	
	void operator * (matrix x){
		int i,j,k;
		matrix t;
		for(i=1;i<=l;i++)
			for(j=1;j<=r;j++)
				t.M[i][j]=0;
		for(i=1;i<=r;i++)
			for(j=1;j<=x.l;j++)
				for(k=1;k<=l;k++)
					t.M[j][i] = (t.M[j][i] + M[k][i] * x.M[j][k]) % Mod;
		for(i=1;i<=l;i++)
			for(j=1;j<=r;j++)
				M[i][j]=t.M[i][j];
	}
}res,c;

void init(){
	int i,j;
	C[0][0]=1;
	for(i=1;i<=50;i++) C[i][0]=C[i][i]=1;
	for(i=1;i<=50;i++)
		for(j=1;j<i;j++) C[i][j]=C[i-1][j]+C[i-1][j-1];
}

int main(){
	int i,j,n;
	ll x;
	init();
	while(scanf("%d%lld%lld",&n,&x,&Mod)){
		if(n==-1&&x==-1&&Mod==-1) break;
		for(i=1;i<=x+12;i++)
			for(j=1;j<=x+2;j++) if(j<=i) c.M[i][j]=(x * C[i-1][j-1]) % Mod; else c.M[i][j]=0;
		
		for(i=1;i<=x;i++) c.M[x+2][i]=0;
		c.M[x+2][x+1]=1; c.M[x+2][x+2]=1;
		c.l=x+2; c.r=x+2;
		for(i=1;i<=x+1;i++) res.M[i][1]=x;
		res.M[x+2][1]=0;
		res.l=x+2; res.r=1;
		for(; n ; n >>= 1){
			if( n & 1){
				res * c;
			}
			c * c;
		}
		printf("%lld\n",res.M[x+2][1]);
	}
	return 0;
}