线段树 CodeForces 220B - Little Elephant and Array

题目链接：http://www.codeforces.com/problemset/problem/220/B

代码风格：www.notonlysuccess.com

题目大意：统计区间内满足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;
}