JZOJ 5439. 【NOIP2017提高A组集训10.31】Calculate

Description

Input

Output

Data Constraint

Solution

• 设前缀和 $g[i][j]$ 表示 $A$ 为 $i$ 、$B%A$ 为 $j$ 的数量。

• 这样就能 $O(1)$ 算出 $t$ 的值，值改变时暴力维护即可。

• 时间复杂度 $O(T*M*MaxAi)$ 。

Code

#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+1,M=1001;
int n,m,mx;
long long num;
long long g[M][M];
int a[N],b[N];
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline void write(long long x){if(x>9) write(x/10);putchar(x%10+'0');}
inline int max(int x,int y){return x>y?x:y;}
inline long long check(long long t)
{
    long long sum=-num,all;
    for(int i=1;i<=mx;i++)
        if(all=g[i][i-1]) sum+=t/i*all-(all-g[i][t%i]);
    return sum;
}
int main()
{
    int T=read();
    while(T--)
    {
        n=read(),m=read(),num=mx=0;
        memset(g,0,sizeof(g));
        for(int i=1;i<=n;i++) mx=max(mx,a[i]=read());
        for(int i=1;i<=n;i++) num+=(b[i]=read())/a[i],g[a[i]][b[i]%a[i]]++;
        for(int i=1;i<=mx;i++)
            for(int j=1;j<=mx;j++) g[i][j]+=g[i][j-1];
        while(m--)
        {
            int type=read();
            if(type==1)
            {
                int x=read(),y=read();
                for(int i=b[x]%a[x];i<a[x];i++) g[a[x]][i]--;
                num-=b[x]/a[x];mx=max(mx,a[x]=y);num+=b[x]/y;
                for(int i=b[x]%y;i<a[x];i++) g[y][i]++;
            }else
            if(type==2)
            {
                int x=read(),y=read();
                for(int i=b[x]%a[x];i<a[x];i++) g[a[x]][i]--;
                num-=b[x]/a[x];b[x]=y;num+=y/a[x];
                for(int i=y%a[x];i<a[x];i++) g[a[x]][i]++;
            }else
            {
                int k=read();
                long long l=0,r=1e12+1e9;
                while(l<r)
                {
                    long long mid=(l+r)>>1;
                    if(check(mid)<k) l=mid+1; else r=mid;
                }
                write(l),putchar('\n');
            }
        }
    }
    return 0;
}