题意:
就是给你一个数组,然后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;
}
总结:
多多思考,多多总结。