牛客月赛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;
}

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

原文地址：https://blog.csdn.net/m0_52398496/article/details/123164772