牛客小白月赛17 ——区间求和(莫队)

在这里插入图片描述

输入描述:
第一行,两个整数n,m
第二行,总共n个数,代表这个数列
接下来m行,每行两个整数l,r,代表一个询问

输出描述:
输出总共m行,对于每个询问,输出这个询问对应的答案

输入

10 5
1 3 2 4 5 6 4 5 6 7
1 5
2 5
3 4
1 10
3 7

输出

15
14
6
73
29

数据范围1-1e5
我之前听过吴巨巨说过莫队,自己也看过csdn,当时没看懂,以为是啥高深的东西,后来写到这个题,看了一下B站上莫队这个算法的讲解,其实挺简单的,事实上莫队就是一个暴力加分块,时间复杂度是n*根号n。
其实这就是一个莫队的裸题,基本上所有莫队主体是相同的,不同的就是它的add函数和sub函数,下面就看代码吧

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
typedef long long ll;
const int mod = 1e8 + 7;
const int MAXN = 1e6 + 10;
const int INF = 0x3f3f3f3f;
ll a[MAXN],ans[MAXN],pos[MAXN],cnt[MAXN];
struct Q{
 int l,r,k;//l:左,r:右,k:第k次询问 
}q[MAXN];
ll res;//res要开ll,否则会爆int 
void add(int x){
    res+=1ll*cnt[a[x]]*a[x]+1ll*(cnt[a[x]]+1)*a[x];//区间加上一个数,那么前面所有与这个数相等的都要加上这个数,然后再加上这个数再乘此时的出现次数 
    cnt[a[x]]++;
}
void sub(int x){
    cnt[a[x]]--;
    res-=1ll*cnt[a[x]]*a[x]+1ll*(cnt[a[x]]+1)*a[x];//区间去掉一个数,这个数的出现次数先减去1,那么前面所有与这个数相等的都要减上这个数,然后再减去这个数再乘此时的出现次数 
}
int main()
{
 int n,m;
 cin>>n>>m;
 int siz=sqrt(n);//分块 
 
 for(int i=1;i<=n;i++)
 {
   cin>>a[i];
   pos[i]=i/siz;//第几块 
 }
  for(int i=1;i<=m;i++)
 {
   cin>>q[i].l>>q[i].r;
  q[i].k=i;
 }
  sort(q+1,q+1+m,[](Q x,Q y){
  return pos[x.l]==pos[y.l]?x.r<y.r:pos[x.l]<pos[y.l];//块相同按右边界排序,块不相同按块大小排序 
 });
 int l=1,r=0;
 for(int i=1;i<=m;i++)
 {
   while(q[i].l<l)add(--l); 
   while(q[i].r>r)add(++r);
   while(q[i].l>l)sub(l++);
   while(q[i].r<r)sub(r--);
   ans[q[i].k]=res;
 }
   for(int i=1;i<=m;i++)
   cout<<ans[i]<<endl;
 } 


谢谢您访问我的博客!

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值