JZOJ 5628. 【NOI2018模拟4.4】Travel

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=\prod_{i=1}^{n}{(a_i+b_i)}$$

• 开 $c$ 棵线段树，代表一个区间代表恰好有 $i$ 个人去A国的方案数。

• 那么有：

  $$ans=\sum_{i=0}^{c-1}f[root][i]$$

• 每次转移时枚举左右子树的 $i$ ，$O(c^2)$ 转移即可。

• 总时间复杂度 $O(Nc^2\ 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;
}