【ZOJ】1610 Count the Colors（线段树查找隔断的线段）

题目大意：有一条最长8000的线段，其中会被最多8000种颜色分割，问一整段是全部颜色的算1，这样每种颜色有多少段呢？

这道题目一开始的思路线段树存放的自然是每一段的情况，设定-1，假如不是-1的话，就表示该段颜色都是相同的，是-1的话，表示下面颜色可以继续分。

只是在查找的时候自己犯难了，想不到怎么去将所有颜色的次数查找出来。

看了题解，这边用的想法是这样的，通过线段树的query功能，每次对分query，直到遇到一段线段是纯色的就不再往下，并且将这段加入事先的数组当中。

最后再遍历这些数组，只要这一条线段与下一条线段的颜色不一样，并且不是相连的，那么对应颜色的次数就++。就是这么暴力，直接！！

曾经竞赛不怕变形题的我，现在只能做做模板题，稍微变形就不会，这真的是因为我做题目没有以前花时间的原因吗？无奈，大学太多事了。

AC代码：

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define MAX 8008
#define ls rt<<1
#define rs ls|1
#define m (l+r)>>1
int sum[MAX << 2];
int a[MAX];
int cnt;
struct pos
{
	int l, r, c;
}p[MAX];
void ups(int rt)
{
	if (sum[rt] != -1)
	{
		sum[ls] = sum[rt];
		sum[rs] = sum[rt];
		sum[rt] = -1;
	}
}

void uprt(int rt)
{
	if (sum[ls] == sum[rs])
		sum[rt] = sum[ls];
}

void updata(int L, int R,int c, int l, int r, int rt)
{
	if (L <= l&&r <= R)
	{
		sum[rt] = c;
		return;
	}
	ups(rt);
	int mid = m;
	if (L < mid)
		updata(L, R, c, l, mid, ls);
	if (mid < R)
		updata(L, R, c, mid, r, rs);
	uprt(rt);
}

void query(int l, int r, int rt)
{
	if (sum[rt] != -1)
	{
		p[cnt].l = l;
		p[cnt].r = r;
		p[cnt++].c = sum[rt];
		return;
	}
	int mid = m;
	if (l == r - 1)return;
	query(l, mid, ls);
	query(mid, r, rs);
}
int main()
{
	int n;
	int L, R,C;
	while (~scanf("%d", &n))
	{
		memset(sum, -1, sizeof(sum));
		memset(a, 0, sizeof(a));
		cnt = 0;
		C = 0;
		int x, y, z;
		for (int i = 0; i < n; i++)
		{
			scanf("%d%d%d", &x, &y, &z);
			updata(x, y, z, 0, MAX, 1);
			C = max(C, z);
		}
		query(0, MAX, 1);
		for (int i = 0; i < cnt; i++)
		{
			if (i != cnt - 1 && p[i].r == p[i + 1].l&&p[i].c == p[i + 1].c)
				continue;
			a[p[i].c]++; 
		}
		for (int i = 0; i <=C;i++)
		if (a[i])
			printf("%d %d\n", i, a[i]);
		cout << endl;
	}
}