题目链接:http://www.codeforces.com/problemset/problem/220/B
题目大意:统计区间内满足a[i] = num[a[i]]的数字有几个
算法:离线算法,线段树
思路:把如果满足a[i] == num[a[i]],那么在它的first位置加一,如果num[a[i]] > a[i],那么把之前的位置-1,并且在新的first位置+1
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1
#define mid int m = (l+r) >> 1
int cnt[454545], sum[454545], next[454545], last[454545], first[454545], num[454545], a[454545], pre[454545], ans[454545];
struct Nod
{
int l, r, id;
Nod(){}
bool operator < (const Nod &cmp) const
{
return r < cmp.r ;
}
}p[456465];
void PushUp(int rt)
{
sum[rt] = sum[rt << 1 |1 ] + sum[rt << 1];
}
void update(int d, int k, int l, int r, int rt)
{
if(l == r)
{
sum[rt] += k;
return ;
}
mid ;
if(d <= m)
update(d, k, lson);
else update(d, k, rson);
PushUp(rt);
}
int to ;
int query(int L, int R, int l, int r, int rt)
{
// printf("%d,%d,%d,%d\n",L, R, l, r);
// scanf("%d", &to);
// printf("rt = %d\n", rt);
if(L <= l && r <= R)
{
return sum[rt];
}
mid ;
int ret = 0;
if(L <= m)
ret += query(L, R, lson);
if(m < R)
ret += query(L, R, rson);
return ret ;
}
int main()
{
int n, m;
while(scanf("%d%d", &n, &m) != EOF)
{
int i;
for(i = 1; i <= n; i ++)
{
scanf("%d", &a[i]);
}
for(i = 0; i < m; i ++)
{
scanf("%d%d", &p[i].l, &p[i].r);
p[i].id = i;
}
sort(p, p+m);
int now = 0;
for(i = 1; i <= n; i ++)
if(a[i] <= n)
{
if(!first[a[i]])
first[a[i]] = i;
cnt[a[i]] ++;
// printf("cnt = %d, a[i] = %d\n", cnt[a[i]], a[i]);
if(last[a[i]])
next[last[a[i]]] = i;
last[a[i]] = i;
if(cnt[a[i]] == a[i])
update(first[a[i]], 1, 1, n, 1);
else if(cnt[a[i]] > a[i])
{
update(first[a[i]], -2, 1, n, 1);
if(pre[a[i]]) update(pre[a[i]], 1, 1, n, 1);
pre[a[i]] = first[a[i]];
first[a[i]] = next[first[a[i]]];
update(first[a[i]], 1, 1, n, 1);
}
while(p[now].r <= i && now < m)
{
// printf("%d%d%d\n", p[now].l, p[now].r, n);
ans[p[now].id] = query(p[now].l, p[now].r, 1, n, 1);
// printf("ans[%d] = %d\n", p[now].id, ans[p[now].id ]);
now ++;
}
}
for(i = 0; i < m; i ++)
{
printf("%d\n", ans[i]);
}
}
return 0;
}