題目鏈接：點擊查看

題目大意：給出一個由n個數字組成的數列，再給出m次查詢，每次查詢要求輸出[l,r]中的答案，這個題目的答案為：

假設x為區間[l,r]內的數，出現的次數記為cnt[x]，則數x的貢獻為cnt[x]^2*x，答案為所有不同數字的貢獻

題目分析：莫隊模板題，直接套模板就行了，對于每增加一個數，那么答案由a^2變為了(a+1)^2=a^2+2*a+1，增加了2*a+1，減少亦是如此，按照這個規則更改一下add函數和del函數就好了

代碼：

#include<iostream> #include<cstdlib> #include<string> #include<cstring> #include<cstdio> #include<algorithm> #include<climits> #include<cmath> #include<cctype> #include<stack> #include<queue> #include<list> #include<vector> #include<set> #include<map> #include<sstream> #include<unordered_map> using namespace std;typedef long long LL;const int inf=0x3f3f3f3f;const int N=1e6+100;int size,n,m,a[N];LL ans[N],cnt[N];struct query {int l,r,id;bool operator<(const query& a)const{if(l/size!=a.l/size)return l<a.l;else if((l/size)&1)return r<a.r;elsereturn r>a.r;} }q[N];LL add(int pos) {cnt[a[pos]]++;return ((cnt[a[pos]]-1)*2+1)*a[pos]; }LL del(int pos) {cnt[a[pos]]--;return -((2*cnt[a[pos]]+1)*a[pos]); }void solve() {int l=1,r=0;LL sum=0;for(int i=1;i<=m;i++){int ql=q[i].l;int qr=q[i].r;while(l<ql)sum+=del(l++);while(l>ql)sum+=add(--l);while(r<qr)sum+=add(++r);while(r>qr)sum+=del(r--);ans[q[i].id]=sum;} }int main() { // freopen("input.txt","r",stdin);ios::sync_with_stdio(false);cin>>n>>m;size=sqrt(n);for(int i=1;i<=n;i++)cin>>a[i];for(int i=1;i<=m;i++){cin>>q[i].l>>q[i].r;q[i].id=i;} sort(q+1,q+1+m);solve();for(int i=1;i<=m;i++)cout<<ans[i]<<endl;return 0; }

?

總結

以上是生活随笔為你收集整理的CodeForces - 86D Powerful array(莫队)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。