Master of Sequence(数学+二分+树状数组)

题意:给两个长度为n的序列(a1,a2......,an)、(b1,b2......bn)和m个询问以及一个整数k,找出满足k <= S(t) = Σ(i=1 to n) ((t-bi) / ai)向下取整  的最小t。考虑函数的单调性此题可以二分,对于该式子可以证明 ((t-bi) / ai)向下取整的值就是 (t/ai)向下取整的值减去 (bi/ai)向下取整的值 如果 (t%ai) - (bi%ai) 的值比0小,说明还得到前面算出来的(t/ai)的值那里拿一个1来减, 所以算出来的值应该减1。 考虑a的值小于1000,用1000个树状数组维护每个(bi % ai)的前缀和,用一个大小为1000的数组nums维护每个ai出现的次数,用一个变量sum记录Σ(i=1 to n)bi/ai的和,则对于每个t,S(t)可以通过以下方式计算得到:i从for 1到1000 记录(t/i) * nums[ai]的值,这个值减去sum得到的是未经过取余处理的值,然后i 从1for到1000的过程中可以减去 (bi %ai)比(t%i)大的个数,这个数可以通过树状数组查询(getsum(t%ai))表示查找<=t%ai的数 然后用总个数nums[ai]减去它得到。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+10;
const int maxm = 1e5+10;
int t,n,m,op;
ll x,y,k;
ll a[maxm],b[maxm];
int nums[maxn];
int lowbit(int x) {return x&(-x);}
struct tree{
    int c[maxn];
    inline void update(int x,int y){
        x++;
        for(int i=x;i<maxn;i+=lowbit(i))
        c[i] += y;
    }
    inline int getsum(int x){
           int ans = 0;x++;
           for(int i=x;i;i-=lowbit(i))
           ans += c[i];
        return ans;
    }
}node[maxn];
bool check(ll t){
    ll tmp = 0;
    for (int i = 1; i <= 1000; i++){
        tmp += (t / i) * nums[i] - (nums[i] - node[i].getsum(t % i));
        if (tmp >= k + 1000 - i) return true;
    }
    return tmp >= k;
}
int main(){
    scanf("%d",&t);
    while (t--){
        ll sum = 0;
        memset(nums, 0, sizeof nums);
        memset(node, 0, sizeof node);        
        scanf("%d%d",&n,&m);
        for (int i = 1; i <= n; i++){
            scanf("%lld",&a[i]);
            nums[a[i]]++;
        }
        for (int i = 1; i <= n; i++){
            scanf("%lld",&b[i]);
            sum += b[i] / a[i];
            node[a[i]].update(b[i]%a[i],1);
        }
        while (m--){
            scanf("%d",&op);
            if (op == 1){
                scanf("%lld%lld",&x,&y);
                nums[a[x]]--,nums[y]++;
                node[a[x]].update(b[x] % a[x], -1);
                node[y].update(b[x] % y, 1);
                sum = sum - b[x] / a[x] + b[x] / y;
                a[x] = y;
            }
            else if (op == 2){
                scanf("%lld%lld",&x,&y);
                node[a[x]].update(b[x]%a[x], -1);
                node[a[x]].update(y%a[x], 1);
                sum = sum - b[x] / a[x] + y / a[x];
                b[x] = y;
            }
            else{
                scanf("%lld",&k);
                ll l = 0, r = 1e12, ans;
                k += sum;
                while (l <= r){
                    ll mid = (l + r) >> 1;
                    if (check(mid)) {
                        ans = mid;
                        r = mid-1;
                    }
                    else l = mid + 1;
                }
                printf("%lld\n",ans);
            }
        }
    }
    return 0;
}