UVa 11235 Frequent values (RMQ)

算法竞赛训练指南，198 页
本题要点：
1、对 int count[MaxN]; //第i段出现的次数
进行RMQ
count 数组的坐标是从1开始的，而数组 d[MaxN][20]则从 0 开始， 在函数RMQ_init 中注意一下
2、根据指南书中的解题思路，关键是求出4个数组出来
int count[MaxN]; //第i段出现的次数
int num[MaxN]; //num[p] 表示位置p所在段的编号
int Left[MaxN]; //Left[p] 表示位置p所在段的左端点
int Right[MaxN]; //Right[p] 表示位置p所在段的右端点

#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MaxN = 100010;
int data[MaxN];
int n, q;
int value[MaxN];	//第i段的数值
int count[MaxN];	//第i段出现的次数
int num[MaxN];		//num[p] 表示位置p所在段的编号
int Left[MaxN];		//Left[p] 表示位置p所在段的左端点
int Right[MaxN];	//Right[p] 表示位置p所在段的右端点
int sum_seg[MaxN];	// sum_seg[i] 表示 1 ~ i 段的所有数的总个数

int cnt = 1;	//段的数量

void init()
{
	cnt = 1;
	value[cnt] = data[1];
	count[cnt] = 1;
	num[cnt] = 1;
	for(int i = 2; i <= n; ++i)
	{
		if(data[i] == data[i - 1])
		{
			count[cnt]++;	
		}else{
			value[++cnt] = data[i];
			count[cnt] = 1;
		}
		num[i] = cnt;
	}
	memset(sum_seg, 0, sizeof(sum_seg));
	for(int i = 1; i <= cnt; ++i)
	{
		sum_seg[i] = sum_seg[i - 1] + count[i];
//		printf("sum_seg[%d] = %d\n", i, sum_seg[i]);
	}
	for(int i = 1; i <= n; ++i)
	{
		int idx = num[i];
		Left[i] = sum_seg[idx - 1] + 1;
		Right[i] = sum_seg[idx];
//		printf("num[%d] = %d, Left[%d] = %d, Right[%d] = %d\n", i, num[i], i, Left[i], i, Right[i]);
	}
}

int d[MaxN][20];	// d[i][j] 表示从 i开始，长度为 2^j 的一段元素的最大值

void RMQ_init()
{
	for(int i = 0; i < cnt; ++i)
	{
		d[i][0] = count[i + 1];		//count 数组的坐标是从1开始的，而数组 d[MaxN][20]则从 0 开始
	}
	for(int j = 1; (1 << j) <= cnt; ++j)
	{
		for(int i = 0; i + (1 << j) - 1 < cnt; ++i)
		{	
			d[i][j] = max(d[i][j - 1], d[i + (1 << (j - 1))][j - 1]);
		}
	}
}

int RMQ(int L, int R)
{
	int k = 0;
	while((1 << (k + 1) <= R - L + 1)) // k满足 2^k <= R - L + 1 的最大整数
	{
		++k;
	}
	return max(d[L][k], d[R - (1 << k) + 1][k]);
}

int query(int L, int R)
{
	if(num[L] == num[R])
	{
		return R - L + 1;
	}else if(num[L] + 1 == num[R]){
		return max(Right[L] - L + 1, R - Left[R] + 1);
	}else{
		int m = max(Right[L] - L + 1, R - Left[R] + 1);
		return max(m, RMQ(num[L] + 1 - 1, num[R] - 1 - 1));	
	}
}

int main()
{
	while(scanf("%d", &n) != EOF && n)
	{
		scanf("%d", &q);
		for(int i = 1; i <= n; ++i)
		{
			scanf("%d", &data[i]);
		}
		init();
		RMQ_init();
		int L, R;
		for(int i = 0; i < q; ++i) // q次查询
		{
			scanf("%d%d", &L, &R);
			printf("%d\n", query(L, R));
		}
	}
	return 0;
}

/*
10 3
-1 -1 1 1 1 1 3 10 10 10
2 3
1 10
5 10
0
*/

/*
1
4
3
*/