大意: $n$个人, 分别属于$m$个组, 要求选出$k$个人, 使得每组至少有一人, 求方案数.

显然答案为$\prod((1+x)^{a_i}-1)$的第$k$项系数, 分治$FFT$即可.

#include <iostream>
#include <cstdio>
#define REP(i,a,n) for(int i=a;i<=n;++i)
#define PER(i,a,n) for(int i=n;i>=a;--i)
#define mid ((l+r)>>1)
#define x first
#define y second
using namespace std;
typedef long long ll;
typedef pair<int*,int> poly;
const int N = 4e5+10, P = 998244353, G = 3, Gi = 332748118;
ll qpow(ll a,ll n) {ll r=1%P;for (a%=P;n;a=a*a%P,n>>=1)if(n&1)r=r*a%P;return r;}
int lim,l,A[N],B[N],R[N];
void init(int n) {
for (lim=1,l=0; lim<=n; lim<<=1,++l) ;
REP(i,0,lim-1) R[i]=(R[i>>1]>>1)|((i&1)<<(l-1));
} void NTT(int *J, int tp=1) {
REP(i,0,lim-1) if (i<R[i]) swap(J[i],J[R[i]]);
for (int j=1; j<lim; j<<=1) {
ll T = qpow(tp==1?G:Gi,(P-1)/(j<<1));
for (int k=0; k<lim; k+=j<<1) {
ll t = 1;
for (int l=0; l<j; ++l,t=t*T%P) {
int y = t*J[k+j+l]%P;
J[k+j+l] = (J[k+l]-y)%P;
J[k+l] = (J[k+l]+y)%P;
}
}
}
if (tp==-1) {
ll inv = qpow(lim, P-2);
REP(i,0,lim-1) J[i]=(ll)inv*J[i]%P;
}
} poly mul(poly a, poly b) {
init(a.y+b.y);
REP(i,0,lim-1) A[i]=B[i]=0;
REP(i,0,a.y) A[i]=a.x[i];
REP(i,0,b.y) B[i]=b.x[i];
NTT(A),NTT(B);
poly c(new int[lim],lim-1);
REP(i,0,lim-1) c.x[i]=(ll)A[i]*B[i]%P;
NTT(c.x,-1);
return c;
} int n,m,k,a[N],fac[N],ifac[N];
int C(int n, int m) {
if (m>n) return 0;
return (ll)fac[n]*ifac[m]%P*ifac[n-m]%P;
}
poly solve(int l, int r) {
if (l==r) {
poly ret(new int[a[l]+1],a[l]);
REP(i,1,a[l]) ret.x[i]=C(a[l],i);
return ret;
}
return mul(solve(l,mid),solve(mid+1,r));
} int main() {
fac[0]=ifac[0]=1;
REP(i,1,N-1) fac[i]=(ll)fac[i-1]*i%P;
ifac[N-1]=qpow(fac[N-1],P-2);
PER(i,1,N-2) ifac[i]=(ll)ifac[i+1]*(i+1)%P;
scanf("%d%d%d", &n, &m, &k);
REP(i,1,m) scanf("%d",a+i);
int ans = solve(1,m).x[k];
if (ans<0) ans += P;
printf("%d\n", ans);
}

最新文章

  1. parawork功能使用说明
  2. 新的一年快开始了,学点新东西吧,从React开始(一)
  3. WebAPI身份验证
  4. ACM:a^b%p-数论-快速幂-快速乘
  5. Git 使用的配置 常用命令
  6. canvas实现钟表
  7. jQuery 的 json 格式的处理问题
  8. rt-thread博客分享
  9. Uva_10253 Series-Parallel Networks
  10. 基于visual Studio2013解决面试题之0610删除重复字符串
  11. oracle报表功能
  12. dotnet core高吞吐Http api服务组件FastHttpApi
  13. Python开发爬虫之静态网页抓取篇:爬取“豆瓣电影 Top 250”电影数据
  14. lombok系列(一)
  15. spring AOP capbilities and goal
  16. Laravel 中使用 JWT 认证的 Restful API
  17. jmeter之正则表达式
  18. 【java编程】加密算法-对称加密及AES加密算法
  19. C#写UTF8文件时指定是否含BOM头
  20. C++中三种传递参数方法的效率分析

热门文章

  1. PHP学习之分页类
  2. vue问题五:element ui组件的开始时间-结束时间验证
  3. 关于Java 8新引入语法特性的简要说明
  4. Splinter自动登录
  5. 阶段5 3.微服务项目【学成在线】_day04 页面静态化_14-页面静态化-数据模型-远程请求接口
  6. 阶段5 3.微服务项目【学成在线】_day03 CMS页面管理开发_16-异常处理-可预知异常处理-自定义异常类型和抛出类
  7. Qt编写自定义控件32-等待进度条控件
  8. 【论文笔记】DeepOrigin: End-to-End Deep Learning for Detection of New Malware Families
  9. 使用 Metrics.net + influxdb + grafana 搭建项目自动化监控和预警方案
  10. Css3 伪元素