【练习】CodeForces86E Powerful array (莫队算法)

最新推荐文章于 2023-03-29 23:41:56 发布

pengwill97

最新推荐文章于 2023-03-29 23:41:56 发布

阅读量164

点赞数

本文链接：https://blog.csdn.net/pengwill97/article/details/81304575

版权

算法---莫队算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

题意

翻译过来大意如下
给定长度为 $n$ 的序列， $t$ 个询问。每个询问给定区间 $[l,r]$ 。并定义 $K_s$ 表示区间内值为 $s$ 的数字个数，请对于每个询问回答 $K_s^2 * s$ 的值。

题解

没有修改操作，考虑离线做法，又因为对于已知区间，可以 $O(1)$ 求出相邻区间，所以用莫队。
我写的比较挫，时间压线过的。。

代码

#include<bits/stdc++.h>
using namespace std;
typedef double db;
typedef long long ll;
typedef unsigned long long ull;
const int nmax = 1e6+7;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const ull p = 67;
const ull MOD = 1610612741;
struct node{
    int l,r,id;
    ll ans;
}Q[nmax];
int n,m;
int a[nmax],belong[nmax],occ[nmax];
ll ans = 0;
bool cmp(node a, node b){
    if(belong[a.l] != belong[b.l]) return a.l < b.l;
    else return a.r < b.r;
}
bool cmpid(node a, node b){
    return a.id < b.id;
}

void modify(int pos,int tag){
    ans = ans - 1LL * occ[a[pos]] * occ[a[pos]] * a[pos];
    occ[a[pos]] += tag;
    ans = ans + 1LL * occ[a[pos]] * occ[a[pos]] * a[pos];
}
int main(){
    scanf("%d %d",&n,&m);
    int sz = sqrt(n);
    for(int i =1;i<=n;++i) scanf("%d",&a[i]);
    for(int i = 1;i<=m;++i){
        scanf("%d %d",&Q[i].l,&Q[i].r), Q[i].id = i;
        belong[i] = (i-1) / sz + 1;
    }
    sort(Q+1,Q+1+m,cmp);
    int l = 1, r = 0;
    for(int i = 1;i<=m;++i){
        while(l<Q[i].l) modify(l,-1),l++;
        while(l>Q[i].l) modify(l-1,1),l--;
        while(r>Q[i].r) modify(r,-1),r--;
        while(r<Q[i].r) modify(r+1,1),r++;
        Q[i].ans = ans;
    }
    sort(Q+1,Q+1+m,cmpid);
    for(int i = 1;i<=m;++i) printf("%I64d\n",Q[i].ans);
    return 0;
}