每日刷题日记3.31

知识点：莫队算法

解答：

//莫队问题求解
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;

typedef struct Node
{
  int id; //第几次的查询
  int l; //左
  int r; //右
  int k; //次数
}Node;//查询节点

const int MAX=1e5+10;
int N,M;

int cnt[MAX]; //数字出现次数
int sum[MAX]; //次数相同之和数组
int a[MAX]; //数字
int ans[MAX]; //查询结果数组
Node nodes[MAX]; //查询结构体数组
int Len; //分成sqrt(N)块

int Com(Node a,Node b) //排序，左区间小排前面，相同则右区间小放前面
{
  if(a.l/Len==b.l/Len) return a.r/Len<b.r/Len;
  else return a.l/Len<b.l/Len;
}
void add(int i)
{
  sum[cnt[a[i]]]--;//总和为cnt[a[i]]的值减一
  cnt[a[i]]++; //a[i]的次数加一
  sum[cnt[a[i]]]++;//总和为cnt[a[i]+1]的值加一
}
void del(int i)
{
  sum[cnt[a[i]]]--;//总和为cnt[a[i]]的值减一
  cnt[a[i]]--; //a[i]的次数减一
  sum[cnt[a[i]]]++;//总和为cnt[a[i]+1]的值加一
}
int main()
{
  cin>>N;
  Len=sqrt(N);
  for(int i=1;i<=N;i++) cin>>a[i];
  cin>>M;
  for(int i=1;i<=M;i++)
  {
    nodes[i].id=i;
    cin>>nodes[i].l>>nodes[i].r>>nodes[i].k;
  }
  sort(nodes+1,nodes+1+N,Com); //排序查询次数,使得复杂度为O(n*sqrt(n))

  int l=1,r=0; //初始化，左指索引l=1,右指索引r=0
  for(int i=1;i<=M;i++)
  {  
    while(l<nodes[i].l) del(l++); 
    while(l>nodes[i].l) add(--l);
    while(r<nodes[i].r) add(++r);
    while(r>nodes[i].r) del(r--);   
    ans[nodes[i].id]=sum[nodes[i].k];
  }
  for(int i=1;i<=M;i++)
    cout<<ans[i]<<endl;
  return 0;
}