codeforces 86D Powerful array 莫队算法

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

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

0 ·

CC 4.0 BY-SA版权

莫队算法专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一道经典的莫队算法模板题目，该题目要求计算指定区间内的元素立方和。文章通过具体的代码实现展示了如何使用莫队算法进行区间查询优化，并详细解释了代码中的关键步骤。

题意：
给你n个数字，m个区间，要你算出每个区间内si*Ksi*ksi总和。
题解：
莫队算法模板题，每移动一次的时候减去之前的影响再加上现在的影响就可以了，具体看代码。

#include<stdio.h> 
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define LL long long int
const int MAXN=200000+7;
const int MAXM=1e6+7;
struct node
{
    int l,r;
    int id;
    int pos;
}q[MAXN];
bool cmp(node c,node d)
{
    if(c.pos==d.pos) return c.r<d.r;
    return c.l<d.l;
}
int block[MAXN];
int a[MAXN],cnt[MAXM];
int n,m;
LL ans[MAXN];
LL val;
void update(int x,int v)
{
    val-=(LL)cnt[a[x]]*cnt[a[x]]*a[x];
    cnt[a[x]]+=v;
    val+=(LL)cnt[a[x]]*cnt[a[x]]*a[x];
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(cnt,0,sizeof(cnt));
        int s=sqrt(n);
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]),block[i]=(i-1)/s+1;
        for(int i=1;i<=m;i++)
        scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i,q[i].pos=block[q[i].l];
        sort(q+1,q+m+1,cmp);
//      for(int i=1;i<=m;i++)
//      printf("%d %d\n",q[i].l,q[i].r); 
        int l=1,r=0;
        val=0;
        for(int i=1;i<=m;i++)
        {
            while(l<q[i].l)
            {
                update(l,-1);
                l++;
            }
            while(l>q[i].l)
            {
                l--;
                update(l,1);
            }
            while(r>q[i].r)
            {
                update(r,-1);
                r--;
            }
            while(r<q[i].r)
            {
                r++;
                update(r,1);
            }
            ans[q[i].id]=val;
        }
        for(int i=1;i<=m;i++)
        printf("%I64d\n",ans[i]);
    }
}