洛谷-2709 小B的询问

最新推荐文章于 2024-11-02 22:09:18 发布

原创最新推荐文章于 2024-11-02 22:09:18 发布 · 153 阅读

0 ·

CC 4.0 BY-SA版权

算法题专栏收录该内容

310 篇文章

订阅专栏

题目描述
小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L…R]，求 $∑(c(i)2)\sum(c(i)^2)$ 的值,其中i的值从1到K，其中c(i)表示数字i在[L…R]中的重复次数。小B请你帮助他回答询问。
输入格式
第一行，三个整数N、M、K。
第二行，N个整数，表示小B的序列。
接下来的M行，每行两个整数L、R。
输出格式
M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

输入输出样例
输入 #1
6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

输出 #1
6
9
5
2

说明/提示
对于全部的数据，1<=N、M、K<=50000

解释：非常简单的莫队算法题，直接上模板

#include<iostream>
#include<cmath>
#include<algorithm>
#define N 50003
#define ll long long
using namespace std;
int pos[N]={0};
ll a[N]={0};
int n=0,m=0,k=0;
ll ret=0;
ll num[N]={0};
struct node{
    int l,r;
    int id;
    ll ans;
}s[N];
bool cmp(node &a,node &b){
    if(pos[a.l]==pos[b.l]){
        return a.r<b.r;
    }
    return a.l<b.l;
}
bool cmp_id(node &a,node &b){
    return a.id<b.id;
}
void update(int p,int v){
    ret-=num[a[p]]*num[a[p]];
    num[a[p]]+=v;
    ret+=num[a[p]]*num[a[p]];
}
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    int mk=sqrt(n);
    for(int i=1;i<=n;i++){
        pos[i]=(i-1)/mk+1;
    }
    for(int i=1;i<=m;i++){
        cin>>s[i].l>>s[i].r;s[i].id=i;
    }
    sort(s+1,s+1+m,cmp);
    for(int i=1,l=1,r=0;i<=m;i++){
        for(;r<s[i].r;r++) update(r+1,1);
        for(;r>s[i].r;r--) update(r,-1);
        for(;l<s[i].l;l++) update(l,-1);
        for(;l>s[i].l;l--) update(l-1,1);
        s[i].ans=ret;
    }
    sort(s+1,s+1+m,cmp_id);
    for(int i=1;i<=m;i++){
        cout<<s[i].ans<<endl;
    }
    return 0;
}