ABC358

ABC358

E - Alphabet Tiles

• 一句話題意：給定 \(K\) 和 \(C_{1\sim 26}\), 問共有多少個長度為 \(1 \sim K\) 的字串，滿足第 \(i\) 種英文字母的出現次數不大於 \(C_i\), 不小於 \(0\).
• 標籤：動態規劃，組合數，動態拆分
• 記 \(f[i][j]\) 表示使用前 \(i\) 種字母，組成長度為 \(j\) 的字串的方案數，考慮如何從 \(f[i-1]\) 轉移到 \(f[i]\).
• \(f[i][j]+=f[i-1][j-k]\times C_{j}^{k}\)
• 有兩點需要注意：
  • 這樣是直接把新的第 \(i\) 種字母的所有貢獻都用組合數算進去了
  • 由於\(j-k\) 個數中可以連續插空，所以係數是 \(C_{j}^{k}\) 而不是 \(C_{j-k}^{k}\).
  • 邊界是 \(f[i][j]=0\).

#include<bits/stdc++.h>
#define F(i,l,r) for(int i(l);i<=r;++i)
#define G(i,r,l) for(int i(r);i>=l;--i)
//#define int long long
using namespace std;
using ll = long long;
const ll mod = 998244353;
const int N=1e3+105;
int n;
int c[N][N],p[30],f[30][N];
inline void ini(){
	F(i,0,n+5) c[i][0]=1;
	F(i,1,n+5) F(j,0,i) c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
signed main(){
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	cin>>n; 
	ini();
	F(i,1,26)  cin>>p[i];
	f[0][0]=1;
	F(i,1,26){
		F(j,1,n){
			F(k,0,min(j,p[i])) {
//				printf("f[%d][%d]=(f[%d][%d]+f[%d][%d]*c[%d][%d])=%d+%d*%d=%lld\n",i,j,i,j,i-1,j-k,j,k,f[i][j],f[i-1][j-k],c[j][k],(f[i][j]+f[i-1][j-k]*c[j][k])%mod);
				f[i][j]=(f[i][j]+f[i-1][j-k]*c[j][k])%mod;				
			}
		}
	}
	int ans=0;
	F(i,1,n) ans=(ans+f[26][i])%mod;
	cout<<ans;
	return 0;
}