Codeforces Round #745 (Div. 2) E. Train Maintenance

Codeforces Round #745 (Div. 2) E. Train Maintenance
补一下这道题，本来以为线段树魔改，结果好像不是，还是自己太菜了，代码主要参考官方solution，然而他的tutorial写的我不太能接受，看完代码还是选择用我自己的方式来解释一遍~
题目大意：有n辆车，m天，每辆车干x天休息y天，每天可以选择一种操作：
操作1：将第k辆车挑出来，开始工作
操作2：第k辆车退休
每天都询问当天有多少辆车在休息。
考虑最朴素的做法，对每个休息区间进行更新，使用差分数组进行维护，答案为前i天的前缀和，这样复杂度为O(n * m)
考虑如下办法：
当x+y >sqrt(m)时，采用上述办法，需要更新的区间个数<=sqrt(m)
当x+y <=sqrt(m)时，可以注意到，假设对列车j开始工作的时间为st,休息区间为[st + x,st + x + y - 1],[st + (x + y) + x ,st + (x + y) + x +y - 1 ] …,即，在模（x+y）时，休息区间统一为[(st + x)%(x + y),(st - 1) % (x + y)],故若当前天数i%（x+y）落在[(st + x)%(x + y),(st - 1) % (x + y)],则该天为列车j的休息日,因此只需要对每个x+y = a维护一下休息区间即可，区间长度<=sqrt(m)
因此，算法总复杂度为O(m ^ 3/2)当m =2e5时，计算次数为9千万左右，勉强能在一秒钟内跑完…这场虽然毒瘤，但还是很有意思的

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
int x[N],y[N];
int a[N];
int s[N];
int thre,ans;
int cnt[505][505];

void update(int st,int k,int v){
	int aa = ( x[k] + y[k]);
	int *c = cnt[aa];
	int l = (st + x[k]) % aa;
	int r = (st  - 1) % aa;
	if(l <= r){
		for(int i = l;i <= r;i++)c[i] += v;
	}else{
		for(int i = 0;i <= r;i++)c[i] += v;
		for(int i = l;i < aa;i++)c[i] += v;
	}
		
}
int query(int x){
	int res = 0;
	for(int i = 2;i <= thre;i++)
		res += cnt[i][x%i];
	return res;
}
int main(){
//	freopen("input.txt","r",stdin);
	int n,m;
	scanf("%d%d",&n,&m);
	thre = sqrt(m);
	for(int i = 1;i <= n;i++){
		scanf("%d%d",&x[i],&y[i]);
	}
	for(int i = 1;i <= m;i++){
		int op,k;
		scanf("%d%d",&op,&k);
		if(op == 1){
			int aa = x[k] + y[k];
			if(aa > thre){
				for(int j = i + x[k];j <= m;j += aa){
					a[j]++;
					if(j + y[k] <= m)a[j + y[k]]--;
				}
			}else{
				update(i,k,1);
			}
			s[k] = i;
		}else{
			int aa = x[k] + y[k];
			if(aa > thre){
				for(int j = s[k] + x[k];j <= m;j += aa){
					a[j]--;
					if(j <= i - 1)ans--;//及时更新ans的差分
					if(j + y[k] <= i - 1)ans++;//及时更新ans的差分
					if(j + y[k] <= m)a[j + y[k]]++;
				}
					
			}else{
				update(s[k],k,-1);
			}
		}
		ans += a[i];
		printf("%d\n",ans + query(i));
	}
	return 0;
}