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

总结:
多多思考,多多总结。

上一篇:Hw06 of CS61A of UCB


下一篇:react 调用api实现数据列表的渲染