codeforces 86 D. Powerful array （莫队算法）

2016-02-13 · 2 min read · 分块莫队算法

http://codeforces.com/problemset/problem/86/D

题意：Ks为区间内s的数目，求区间[L,R]之间所有KsKss的和

思路：莫队算法，和小z的袜子差不多。不明白第一次tle#54是什么情况。把每一块的大小改成了常数之后就过了。

再交一遍就过了。。不过貌似根据最大数据把siz大小设置成一个常数比根号n要块很多==

/* ***********************************************
Author :111qqz
Created Time :2016年02月13日 星期六 23时17分58秒
File Name :code/cf/problem/86D.cpp
************************************************ */

 1#include <cstdio>
 2#include <cstring>
 3#include <iostream>
 4#include <algorithm>
 5#include <vector>
 6#include <queue>
 7#include <set>
 8#include <map>
 9#include <string>
10#include <cmath>
11#include <cstdlib>
12#include <ctime>
13#define fst first
14#define sec second
15#define lson l,m,rt<<1
16#define rson m+1,r,rt<<1|1
17#define ms(a,x) memset(a,x,sizeof(a))
18typedef long long LL;
19#define pi pair < int ,int >
20#define MP make_pair

 1using namespace std;
 2const double eps = 1E-8;
 3const int dx4[4]={1,0,0,-1};
 4const int dy4[4]={0,-1,1,0};
 5const int inf = 0x3f3f3f3f;
 6const int N=2E5+7;
 7int t,n;
 8int a[N];
 9int pos[N];
10LL sum;
11LL ans[N];
12struct node
13{
14    int l,r;
15    int id;

1    bool operator <(node b)const
2    {
3    if (pos[l]==pos[b.l]) return r<b.r;
4    return pos[l]<pos[b.l];
5    }

}q[N];

int cnt[N*5];

 1void update(int x,int d)
 2{
 3    sum -= 1LL*cnt[a[x]]*cnt[a[x]]*a[x];
 4    cnt[a[x]]+=d;
 5    sum += 1LL*cnt[a[x]]*cnt[a[x]]*a[x];
 6}
 7int main()
 8{
 9    #ifndef  ONLINE_JUDGE 
10    freopen("code/in.txt","r",stdin);   //TLE #54....WHY?
11  #endif

1    cin>>n>>t;
2    int bk = 470;
3    for ( int i =  1 ; i <= n ; i++)
4    {
5        scanf("%d",&a[i]);
6        pos[i]=(i-1)/bk;
7    }

    for ( int i = 1 ; i <= t ; i++) scanf("%d %d",&q[i].l,&q[i].r),q[i].id = i ;
    sort(q+1,q+t+1);

 1    int pl = 1;
 2    int pr = 0;
 3    int id;
 4    int l;
 5    int r;
 6    ms(cnt,0);
 7    sum = 0;
 8    for ( int i = 1 ; i <= t ; i++)
 9    {
10     //   cout<<"sum:"<<sum<<endl;
11        id = q[i].id;
12        l = q[i].l;
13        r = q[i].r;

 1        if (pr<r)
 2        {
 3        for ( int j = pr+1 ; j <= r ; j++)
 4            update(j,1);
 5        }
 6        else
 7        {
 8        for ( int j = r+1 ; j <= pr ; j++)
 9            update(j,-1);
10        }
11        pr = r;

 1        if (pl<l)
 2        {
 3        for ( int j = pl ; j <=l-1 ; j++)
 4            update(j,-1);
 5        }
 6        else
 7        {
 8        for ( int j = l ; j <= pl-1 ; j++)
 9            update(j,1);
10        }

        pl = l;

        ans[id] = sum;
    }

    for ( int i = 1 ;i <= t ; i++) printf("%lld\n",ans[i]);

1  #ifndef ONLINE_JUDGE  
2  fclose(stdin);
3  #endif
4    return 0;
5}