Description
有N个人出去旅行,第i个人去A国有Ai种游玩方式,去B国有Bi种游玩方式,问至少有C个人去A国的情况下,所有人的游玩方式有多少种不同的可能。
两种所有人的游玩方式不同当且仅当存在一个人选择的游玩方式不同,或选择去的国家不同。
接下来有P次修改,每次修改一个人的Ai和Bi。
Input
第一行两个正整数,表示N,C,含义如题所示。
接下来一行N个整数,第i个整数表示Ai。
接下来一行N个整数,第i个整数表示Bi。
接下来一行一个正整数表示P。
接下来P行,每行三个整数i,x,y,表示修改Ai为x,Bi为y。
Output
对每次修改输出一行一个整数,表示总方案数 mod 10007。
Sample Input
4 2
1 2 3 4
1 2 3 4
1
4 1 1
Sample Output
66
Data Constraint
对于100%的数据满足:Ai,Bi,x,y在int范围内。
Solution
直接线段树……
由于 c 特别的小,考虑正难则反。
即用总方案数
tot ,减去有 0 到c−1 个人到A国的方案数 ans 。易得:
tot=∏i=1n(ai+bi)开 c 棵线段树,代表一个区间代表恰好有
i 个人去A国的方案数。那么有:
ans=∑i=0c−1f[root][i]每次转移时枚举左右子树的 i ,
O(c2) 转移即可。总时间复杂度 O(Nc2 logN) 。
Code
#include<cstdio>
#include<cctype>
using namespace std;
const int N=1e5+5,C=20,mo=10007;
int n,c,qi,tot;
int f[N<<2][C],a[N],b[N];
inline int read()
{
int X=0,w=0; char ch=0;
while(!isdigit(ch)) w|=ch=='-',ch=getchar();
while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
return w?-X:X;
}
inline void write(int x)
{
if(x>9) write(x/10);
putchar(x%10+'0');
}
inline int ksm(int x,int y)
{
int s=1;
while(y)
{
if(y&1) s=s*x%mo;
x=x*x%mo;
y>>=1;
}
return s;
}
inline void update(int v)
{
int ls=v<<1,rs=ls|1;
for(int i=0;i<c;i++) f[v][i]=0;
for(int i=0;i<c;i++)
for(int j=0;i+j<c;j++)
f[v][i+j]=(f[v][i+j]+f[ls][i]*f[rs][j])%mo;
}
void make(int v,int l,int r)
{
if(l==r)
{
f[v][0]=b[l];
f[v][1]=a[l];
return;
}
int mid=l+r>>1;
make(v<<1,l,mid);
make(v<<1|1,mid+1,r);
update(v);
}
void change(int v,int l,int r)
{
if(l==r)
{
f[v][0]=b[qi];
f[v][1]=a[qi];
return;
}
int mid=l+r>>1;
if(qi<=mid) change(v<<1,l,mid); else change(v<<1|1,mid+1,r);
update(v);
}
int main()
{
n=read(),c=read();
for(int i=1;i<=n;i++) a[i]=read()%mo;
for(int i=tot=1;i<=n;i++)
{
b[i]=read()%mo;
tot=tot*(a[i]+b[i])%mo;
}
make(1,1,n);
int p=read();
while(p--)
{
qi=read();
int x=read()%mo,y=read()%mo;
tot=tot*ksm(a[qi]+b[qi],mo-2)%mo;
tot=tot*(x+y)%mo;
a[qi]=x,b[qi]=y;
change(1,1,n);
int ans=0;
for(int i=0;i<c;i++) ans=(ans+f[1][i])%mo;
ans=(tot-ans+mo)%mo;
write(ans),putchar('\n');
}
return 0;
}