【CodeForces】【莫队算法】problem 86 D Powerful array 2018年8月20日组队训练

xiaoxiao2021-03-01  4

【题目】http://codeforces.com/problemset/problem/86/D

【题意】给出一个长度为n的串,t次询问,每次询问l到r之间每个数*这个数出现的次数^2的和

【思路】莫队算法

【代码】

#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=2e5+5;//区间范围 const int MAX=1e6+5;//最大数字 int unit,cnt[MAX],arr[N]; ll res[N];//存答案 ll ans=0;//当前的答案 struct node { int l,r,id; } q[N]; bool cmp(node a,node b)//对询问排序 { return a.l/unit!=b.l/unit?a.l/unit<b.l/unit:a.r<b.r; } void add(int pos)//增加 { ans-=1ll*cnt[arr[pos]]*cnt[arr[pos]]*arr[pos];//减去旧的 cnt[arr[pos]]++; ans+=1ll*cnt[arr[pos]]*cnt[arr[pos]]*arr[pos];//加上新的 } void remove(int pos)//减少 { ans-=1ll*cnt[arr[pos]]*cnt[arr[pos]]*arr[pos];//减去旧的 cnt[arr[pos]]--; ans+=1ll*cnt[arr[pos]]*cnt[arr[pos]]*arr[pos];//加上新的 } int main() { int n; scanf("%d",&n); int m; scanf("%d",&m); unit=sqrt(n);//分块 for(int i=1; i<=n; i++) { scanf("%d",&arr[i]); } for(int i=1; i<=m; i++) { scanf("%d%d",&q[i].l,&q[i].r); q[i].id=i; } sort(q+1,q+m+1,cmp); int L=q[1].l,R=L-1; for(int i=1; i<=m; i++) { while(L>q[i].l) add(--L); while(L<q[i].l) remove(L++); while(R>q[i].r) remove(R--); while(R<q[i].r) add(++R); res[q[i].id]=ans; } for(int i=1; i<=m; i++) { printf("%lld\n",res[i]); } }

 

转载请注明原文地址: https://www.6miu.com/read-3850121.html

最新回复(0)