CodeForces - 86D Powerful array

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

原创最新推荐文章于 2023-03-29 23:41:56 发布 · 310 阅读

0 ·

CC 4.0 BY-SA版权

莫队算法专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个使用莫队算法解决特定区间查询问题的例子。通过详细解释代码实现，展示了如何高效地处理一系列区间内的元素操作和查询任务。

题目：给你n个数,m个询问，Ks为区间[l,r]中s出现的次数，求区间[l,r]中所有的Ks*Ks*s的和。

思路：莫队算法

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define PI acos(-1.0)
#define LL __int64
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f

const int maxn=1e6+50;
struct Ask{
    int L,R,id;
}ask[maxn];
LL ans[maxn];
int arr[maxn],num[maxn];//从1开始计数
int n,m,siz;//n个数，m个询问，块的大小
bool cmp(Ask a,Ask b){
    if(a.L/siz!=b.L/siz) return a.L/siz<b.L/siz;
    else return a.R<b.R;
}
void solve(){
    LL temp=0;
    mm(num,0);
    int L=1,R=0;
    for(int i=1;i<=m;i++){
        while(R<ask[i].R){
            R++;
            temp-=(LL)num[arr[R]]*num[arr[R]]*arr[R];
            num[arr[R]]++;
            temp+=(LL)num[arr[R]]*num[arr[R]]*arr[R];
        }
        while(R>ask[i].R){
            temp-=(LL)num[arr[R]]*num[arr[R]]*arr[R];
            num[arr[R]]--;
            temp+=(LL)num[arr[R]]*num[arr[R]]*arr[R];
            R--;
        }
        while(L<ask[i].L){
            temp-=(LL)num[arr[L]]*num[arr[L]]*arr[L];
            num[arr[L]]--;
            temp+=(LL)num[arr[L]]*num[arr[L]]*arr[L];
            L++;
        }
        while(L>ask[i].L){
            L--;
            temp-=(LL)num[arr[L]]*num[arr[L]]*arr[L];
            num[arr[L]]++;
            temp+=(LL)num[arr[L]]*num[arr[L]]*arr[L];
        }
        ans[ask[i].id]=temp;
    }
}
int main(){

    while(~scanf("%d%d",&n,&m)){
        for(int i=1;i<=n;i++)
            scanf("%d",&arr[i]);
        for(int i=1;i<=m;i++){
            ask[i].id=i;
            scanf("%d%d",&ask[i].L,&ask[i].R);
        }
        siz=(int)sqrt(n*1.0);
        sort(ask+1,ask+m+1,cmp);
        solve();
        for(int i=1;i<=m;i++)
            printf("%I64d\n",ans[i]);
    }
    return 0;
}