分块80分求调
查看原帖
分块80分求调
785917
Chthollian楼主2024/10/15 16:38
#include<bits/stdc++.h>
using namespace std;
int n,f,len,id[100005],op,l,r,fir[100005],lst[100005];
long long blo[100005][2],a[100005],tag[100005],k;
void change(int l,int r)
{
	int idl=id[l],idr=id[r];
	if(idl==idr)
	{
		for(int i=l; i<=r; i++)
		{
			blo[idl][a[i]]--;
			a[i]^=1;
			blo[idl][a[i]]++;
		}
		return;
	}
	for(int i=l; i<=lst[idl]; i++)
	{
		blo[id[i]][a[i]]--;
		a[i]^=1;
		blo[id[i]][a[i]]++;
	}
	for(int i=fir[idr]; i<=r; i++)
	{
		blo[id[i]][a[i]]--;
		a[i]^=1;
		blo[id[i]][a[i]]++;
	}
	for(int i=idl+1; i<idr; i++)
	{
		tag[i]^=1;
	}
}
long long ask(int l,int r)
{
	long long ans=0;
	int idl=id[l],idr=id[r];
	if(idl==idr)
	{
		for(int i=l; i<=r; i++)
		{
			ans+=a[i]^tag[id[i]];
		}
		return ans;
	}
	for(int i=l; i<=lst[idl]; i++)
	{
		ans+=a[i]^tag[id[i]];
	}
	for(int i=fir[idr]; i<=r; i++)
	{
		ans+=a[i]^tag[id[i]];
	}
	for(int i=idl+1; i<idr; i++)
	{
		ans+=blo[i][1^tag[i]];
	}
	return ans;
}
int main()
{
	cin>>n>>f;
	len=sqrt(n);
	for(int i=1; i<=n; i++)
	{
		char c;
		cin>>c;
		a[i]=c-'0';
		id[i]=(i-1)/len+1;
		blo[id[i]][a[i]]++;
	}
	for (int i=1; i<=n; i++)
	{
		if (id[i]!=id[i-1])
		{
			fir[id[i]]=i,lst[id[i-1]]=i-1;
		}
	}
	lst[id[n]]=n;
	while(f--)
	{
		cin>>op>>l>>r;
		if(op==0)
		{
			change(l,r);
		}
		else
		{
			cout<<ask(l,r)<<"\n";
		}
	}
	return 0;
}
2024/10/15 16:38
加载中...