codeforces problem 140E New Year Garland

排列组合题
题意
用m种颜色的彩球装点n层的圣诞树。圣诞树的第i层恰由l[i]个彩球串成一行,且同一层内的相邻彩球颜色不同,同时相邻两层所使用彩球的颜色集合不同。求有多少种装点方案,答案对p取模。
只要任一位置上的彩球颜色不同,就算作不同的方案。
1 <= n,m <= 10^6, 2 <= p <= 10^9, 1 <= l[i] <= 5000, Σl[i] <= 10^7
若只考虑一层,就是bzoj1008的越狱,但是要考虑上下两层不同,卒。
考虑只在一层中,使用前j种颜色,前i个位置。
不是很好写,引入了最小表示法的思想,使第j种颜色一定要出现于第j-1种后才行
f[i][j]=f[i-1][j](j-1)+f[i-1][j-1];
这样表示的话会少了排列 因此算方案时只要
j!即可

然后考虑一层层算
d[i][j]表示第i层用了j种颜色(不包括前i-1层的颜色)
那么 就有
d[i][j]=C(j,m)j!f[l[i]][j]Σd[i-1][k] -x(x为冲突情况情况数)
因为d[i-1][j]中已经计算了排列和组合
因此不用再乘组合数 所以x=d[i-1][j]
f[l[i]][j]*j!
因此解决

转载于:https://www.cnblogs.com/wkingG/p/5260498.html

你可能感兴趣的:(codeforces problem 140E New Year Garland)