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