CodeForces 86 D Powerful array 莫队

题意：求区间[l, r] 内的数的出现次数的平方 * 该数字。

题解：莫队离线操作，然后加减位置的时候直接修改答案就好了。

这个题目中发现了一个很神奇的事情，本来数组开1e6大小就直接过了4100+ms，想测试一下inline，顺手把空间砍成了刚好够用，然后跑的更慢了 4700+ms，删了inline之后T了，到现在也不知道发生了啥。

如果有大牛路过希望帮我看下， CF run id 38822420（4100+） 38822607（4700+）38822636（TLE）。然后就是我空间稍微开大一点就跑的更快了。

代码：

 #include<bits/stdc++.h>

 using namespace std;

 #define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);

 #define LL long long

 #define ULL unsigned LL

 #define fi first

 #define se second

 #define pb push_back

 #define lson l,m,rt<<1

 #define rson m+1,r,rt<<1|1

 #define max3(a,b,c) max(a,max(b,c))

 #define min3(a,b,c) min(a,min(b,c))

 #define _S(X) cout << x << ' ';

 #define __S(x) cout << x << endl;

 typedef pair<int,int> pll;

 const int INF = 0x3f3f3f3f;

 const LL mod =  (int)1e9+;

 const int N = 8e6+ ;

 int n, m, blo;

 int cnt[N];

 int a[N];

 LL ans[N];

 LL tmp;

 struct Node{

     int l, r;

     int id;

 }q[N];

 void Add(int p){

     tmp -= 1ll * a[p] * cnt[a[p]] * cnt[a[p]];

     cnt[a[p]]++;

     tmp += 1ll * a[p] * cnt[a[p]] * cnt[a[p]];

 }

 void Remove(int p){

     tmp -= 1ll * a[p] * cnt[a[p]] * cnt[a[p]];

     cnt[a[p]]--;

     tmp += 1ll * a[p] * cnt[a[p]] * cnt[a[p]];

 }

 bool cmp(Node x1, Node x2){

     if(x1.l/blo != x2.l/blo)

         return x1.l/blo < x2.l / blo;

     return x1.r < x2.r;

 }

 int main(){

     scanf("%d%d", &n, &m);

     blo = sqrt(n);

     for(int i = ; i <= n; i++)

         scanf("%d", &a[i]);

     for(int i = ; i <= m; i++){

         scanf("%d%d", &q[i].l, &q[i].r);

         q[i].id = i;

     }

     sort(q+, q++m, cmp);

     int nL = , nR = , tL, tR;

     for(int i = ; i <= m; i++){

         tL = q[i].l, tR = q[i].r;

         while(nL < tL) Remove(nL++);

         while(nL > tL) Add(--nL);

         while(nR <= tR) Add(nR++);

         while(nR > tR+) Remove(--nR);

         ans[q[i].id] = tmp;

     }

     for(int i = ; i <= m; i++){

         printf("%I64d\n", ans[i]);

     }

     return ;

 }

86D

巴特西

CodeForces 86 D Powerful array 莫队

最新文章

热门文章