牛客月赛17-区间求和-(莫队)

G

题意:
就是给你一个数组，然后m次询问，每次询问一段区间，然后让你求出，这个区间的每个数*这个数出现的次数之和。

思考：
刚开始看到想了很久，感觉就是个简单的区间处理问题啊，但是怎么都想不到怎么去维护这个答案，最后看了题解才知道是莫队，然后就去学了下莫队。就是先对每个点都进行分块，看看每个点是在那个分块里面的。然后呢就是暴力处理区间问题，暴力的手段就是对每次处理的区间排序，一点一点的操作，也就是去离线操作，先把需要查询的东西排序，尽可能的少移动，然后维护出答案就行。main函数里面的操作一般都是板子题，主要是理解题意和处理好add和sub函数就行了。
莫队入门题目：小B的询问
今天又做到一个莫队：水宝宝的数据结构题，反正本质也是一样的，唯一多点的操作就是每次查询的是两个区间，稍微操作一下就行了。

代码：

本题代码：

struct Node{
	int l,r;
	int id;
}node[N];

int T,n,m,k;
int ans;
int va[N];
int pos[N];
int cnt[N];
int anw[N];

bool cmp(Node A,Node B)
{
	if(pos[A.l]!=pos[B.l]) return pos[A.l]<pos[B.l];
	return A.r<B.r;	
}

void add(int x)
{
	ans -= cnt[va[x]]*cnt[va[x]]*va[x];
	cnt[va[x]] += 1;
	ans += cnt[va[x]]*cnt[va[x]]*va[x];
}

void sub(int x)
{
	ans -= cnt[va[x]]*cnt[va[x]]*va[x];
	cnt[va[x]] -= 1;
	ans += cnt[va[x]]*cnt[va[x]]*va[x];
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>va[i];
	int siz = sqrt(n);
	for(int i=1;i<=n;i++) pos[i] = i/siz;
	for(int i=1;i<=m;i++)
	{
		int l,r;
		cin>>l>>r;
		node[i] = {l,r,i};
	}
	sort(node+1,node+1+m,cmp);
	int l = 1,r = 0;
	for(int i=1;i<=m;i++)
	{
		while(node[i].l<l) add(--l);
		while(node[i].r>r) add(++r);
		while(node[i].l>l) sub(l++);
		while(node[i].r<r) sub(r--);
		anw[node[i].id] = ans;
	}
	for(int i=1;i<=m;i++) cout<<anw[i]<<"\n";
	return 0;
}

小B的询问：
struct Node{
	int l,r;
	int id;
}node[N];

int T,n,m,k;
int ans;
int va[N];
int pos[N];
int cnt[N];
int anw[N];

bool cmp(Node A,Node B)
{
	if(pos[A.l]!=pos[B.l]) return pos[A.l]<pos[B.l];
	return A.r<B.r;
}

void add(int x)
{
	ans -= cnt[va[x]]*cnt[va[x]];
	cnt[va[x]] += 1;
	ans += cnt[va[x]]*cnt[va[x]];	
}

void sub(int x)
{
	ans -= cnt[va[x]]*cnt[va[x]];
	cnt[va[x]] -= 1;
	ans += cnt[va[x]]*cnt[va[x]];
}

signed main()
{
	IOS;
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++) cin>>va[i];
	int siz = sqrt(n);
	for(int i=1;i<=n;i++) pos[i] = i/siz;
	for(int i=1;i<=m;i++)
	{
		int l,r;
		cin>>l>>r;
		node[i] = {l,r,i};
	}
	sort(node+1,node+1+m,cmp);
	int l = 1,r = 0;
	for(int i=1;i<=m;i++)
	{
		while(node[i].l<l) add(--l);
		while(node[i].r>r) add(++r);
		while(node[i].l>l) sub(l++);
		while(node[i].r<r) sub(r--);
		anw[node[i].id] = ans;
	}
	for(int i=1;i<=m;i++) cout<<anw[i]<<"\n";
	return 0;
}

水宝宝的数据结构题：
struct Node{
	int l,r,x;
	int id;
}node[N];

int T,n,m,k;
int va[N];
int cnt[N];
int pos[N];
int idx;
PII anw[N];

bool cmp(Node A,Node B)
{
	if(pos[A.l]!=pos[B.l]) return pos[A.l]<pos[B.l];
	return A.r<B.r;
}

void add(int x)
{
	cnt[va[x]]++;
}

void sub(int x)
{
	cnt[va[x]]--;
}


signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>va[i];
	int siz = sqrt(n);
	for(int i=1;i<=n;i++) pos[i] = i/siz;
	for(int i=1;i<=m;i++)
	{
		int l1,r1,l2,r2,x;
		cin>>l1>>r1>>l2>>r2>>x;
		if(l1>r1) swap(l1,r1);
		if(l2>r2) swap(l2,r2);
		node[++idx] = {l1,r1,x,i};
		node[++idx] = {l2,r2,x,i+m};
	}
	sort(node+1,node+1+idx,cmp);
	int l = 1,r = 0;
	for(int i=1;i<=idx;i++)
	{
		while(node[i].l<l) add(--l);
		while(node[i].r>r) add(++r);
		while(node[i].l>l) sub(l++);
		while(node[i].r<r) sub(r--);
		int id = node[i].id;
		if(id<=m) anw[id].fi = cnt[node[i].x];
		else anw[id-m].se = cnt[node[i].x];
	}
	for(int i=1;i<=m;i++)
	{
		int sum = anw[i].fi*anw[i].se%mod;
		cout<<anw[i].fi<<"\n"<<anw[i].se<<"\n"<<sum<<"\n";
	}
	return 0;
}

总结：
多多思考，多多总结。