RELATIVNOST —— 线段树优化

202 篇文章 5 订阅
96 篇文章 1 订阅

应该能进得去

题意:

现在有n个人要买你的画,第i个人最多买ai个彩色的画,bi个黑白的画,你现在想要至少有c个人买了彩色的画,接下来有q个修改,每次修改某个人的ai和bi。问你每次修改之后有多少种情况可以满足你的要求。

题解:

又是一道线段树优化的dp,因为c只有20所以我们将题目变成:所有情况-买你彩色的画的人的个数少于c个的情况。
也就是 ∏ i = 1 n ( a [ i ] + b [ i ] ) − ∑ i = 0 c − 1 d p [ n ] [ i ] \prod_{i=1}^{n}(a[i]+b[i])-\sum_{i=0}^{c-1}dp[n][i] i=1n(a[i]+b[i])i=0c1dp[n][i]
dp存的就是到第i个人的时候,有c个人买了你的画。
那么状态转移方程就是 d p [ i ] [ j ] = d p [ i − 1 ] [ j ] ∗ a i + d p [ i − 1 ] [ j − 1 ] ∗ b i dp[i][j]=dp[i-1][j]*ai+dp[i-1][j-1]*bi dp[i][j]=dp[i1][j]ai+dp[i1][j1]bi
但是每次都有一个修改,这样的话都要重新再做一遍,时间复杂度肯定是不够的。但是我们发现这个dp没有先后关系的限制,那么就可以用线段树优化去做,这样子每次修改的时间都是 O ( l o g n ∗ 200 ) O(logn*200) O(logn200)
线段树向上更新的时候只需要每次枚举左子树买了的个数和右子树买了的个数:

ans[root][i+j]=(ans[root][i+j]+ans[root<<1][i]*ans[root<<1|1][j])%mod;

然后总的情况数的话,要注意它的模数是10007,所以会有很多数是它的倍数,因此无法逆元,所以我们要记录一下当前数是否有它的倍数,如果有,那么总的情况数就是0.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+1,M=21,mod=10007;
int a[N],b[N],ans[N*4][M];
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
int n,c;
void push_up(int root){
    for(int i=0;i<c;i++)ans[root][i]=0;
    for(int i=0;i<c;i++){
        if(!ans[root<<1][i])continue;
        for(int j=0;j<c-i;j++){
            if(!ans[root<<1|1][j])continue;
            ans[root][i+j]=(ans[root][i+j]+ans[root<<1][i]*ans[root<<1|1][j])%mod;
        }
    }
}
void build(int l,int r,int root){
    if(l==r){
        ans[root][0]=b[l]%mod;
        ans[root][1]=a[l]%mod;
        return ;
    }
    int mid=l+r>>1;
    build(l,mid,root<<1);
    build(mid+1,r,root<<1|1);
    push_up(root);
}
void update(int l,int r,int root,int p){
    if(l==r){
        ans[root][0]=b[l]%mod;
        ans[root][1]=a[l]%mod;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=p)
        update(l,mid,root<<1,p);
    else
        update(mid+1,r,root<<1|1,p);
    push_up(root);
}
int main()
{
    ll p=1;
    int f=0;
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
        scanf("%d",&b[i]);
        if((a[i]+b[i])%mod==0)
            f++;
        else
            p=p*(a[i]+b[i])%mod;
    }

    build(1,n,1);
    int q;
    scanf("%d",&q);
    while(q--){
        int x;
        scanf("%d",&x);
        
        if((a[x]+b[x])%mod==0)
            f--;
        else
            p=p*qpow(a[x]+b[x],mod-2)%mod;
            
        scanf("%d%d",&a[x],&b[x]);
        
        if((a[x]+b[x])%mod==0)
            f++;
        else
            p=p*(a[x]+b[x])%mod;
            
        //if((a[x]))
        update(1,n,1,x);
        ll s=0;
        for(int i=0;i<c;i++)
            s=(s+ans[1][i])%mod;
        printf("%d\n",((f?0:p)-s+mod)%mod);
    }
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值