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; }