当前位置: 代码迷 >> 综合 >> 【Yandex.Algorithm 2011 Round 2, problem: (D) Powerful array】 莫队算法
  详细解决方案

【Yandex.Algorithm 2011 Round 2, problem: (D) Powerful array】 莫队算法

热度:93   发布时间:2023-12-29 02:47:09.0

http://codeforces.com/problemset/problem/86/D
题意就是每种数字x对答案的贡献是(x*x*出现次数),所以add函数和del函数就很明显了。
但是由于读入比较多,需要挂个简单的读入挂
代码

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn  = 1e6+5;
int a[maxn];
long long  sum[maxn];
int pos[maxn];
long long ans[maxn];
struct data
{int l,r,id;
}Q[maxn];
bool cmp(const data &a,const data &b)
{if(pos[a.l]==pos[b.l]) return a.r<b.r;return pos[a.l]<pos[b.l];
}
int L=0,R=0;
long long Ans=0;
void add(int x)
{sum[a[x]]++;Ans+=(1LL*(2LL*sum[a[x]]-1)*a[x]);//算出简单的转移减少常数
}
void del(int x)
{sum[a[x]]--;Ans-=(1LL*(2LL*sum[a[x]]+1)*a[x]);
}
int read()
{ll x=0;char c=getchar();while(c < '0' || c > '9'){c = getchar();}while(c >= '0' && c <= '9'){x = x * 10 + c - '0';c = getchar();}return x;
}
int main()
{int n,t;scanf("%d%d",&n,&t);int sz=sqrt(n);for(int i=1;i<=n;i++){a[i]=read();pos[i]=i/sz;}for(int i=1;i<=t;i++){Q[i].l=read();Q[i].r=read();Q[i].id=i;}sort(Q+1,Q+1+t,cmp);for(int i=1;i<=t;i++){while(L>Q[i].l){L--;add(L);}while(R<Q[i].r){R++;add(R);}while(L<Q[i].l){del(L);L++;}while(R>Q[i].r){del(R);R--;}ans[Q[i].id]=Ans;}for(int i=1;i<=t;i++)  printf("%I64d\n",ans[i]);return 0;
}

友情推荐莫队算法专题链接

  相关解决方案