CF86D Powerful array

题意:t次询问区间[l,r]xax×cnt[ax]2cnt为出现次数

裸的莫队题,用cnt[x]表示x出现的次数,根据平方的性质,我们得出加减操作

  • 加:ans+=ax×2cnt[ax]+1,cnt[ax]++
  • 减:ans=ax×2cnt[ax]1,cnt[ax]

复杂度仍然是O(nn)

Code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 2000000
using namespace std;
struct node
{
    int l,r,b,id;
}q[N+5];
int n,m,a[N+5],blo;
long long ans[N+5],tmp,cnt[N+5];
int cmp(node x,node y)     //奇偶性排序
{
    return (x.b==y.b)?(x.b%2==1?x.r<y.r:x.r>y.r):x.l<y.l;
}
void add(int x)
{
    tmp+=(long long)a[x]*1LL*(cnt[a[x]]*2+1);
    cnt[a[x]]++;
}
void del(int x)
{
    tmp-=(long long)a[x]*1LL*(cnt[a[x]]*2-1);
    cnt[a[x]]--;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    blo=(sqrt(m*2/3))?n/sqrt(m*2/3):sqrt(n);  //块大小
    if (!blo)blo=sqrt(n);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].b=q[i].l/blo;
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    for (int i=1,l=1,r=0;i<=m;i++)    //莫队
    {
        while (l>q[i].l)add(--l);
        while (r<q[i].r)add(++r);
        while (l<q[i].l)del(l++);
        while (r>q[i].r)del(r--);
        ans[q[i].id]=tmp;
    }
    for (int i=1;i<=m;i++)
        printf("%lld\n",ans[i]);
    return 0;
}
posted @   eee_hoho  阅读(51)  评论(0编辑  收藏  举报
编辑推荐:
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· .NET 9 new features-C#13新的锁类型和语义
阅读排行:
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 《HelloGitHub》第 106 期
· 数据库服务器 SQL Server 版本升级公告
· 深入理解Mybatis分库分表执行原理
· 使用 Dify + LLM 构建精确任务处理应用
点击右上角即可分享
微信分享提示