【牛客 - 157C】PH试纸（前缀和，或权值线段树，主席树）

题干：

链接：https://ac.nowcoder.com/acm/contest/157/C
来源：牛客网

题目描述

    PH试纸，是一种检测酸碱度的试纸，试纸红色为酸性，蓝色为碱性。

    HtBest有一个PH试纸，试纸被分成了n段，每一段都可以被染色成红色或者蓝色，WHZ在试纸的每一段上都染为一种颜色，HtBest有m个询问，对于每个询问，Ta想知道某种颜色第qi次在什么地方出现。

输入描述:

第一行有两个正整数n，m。
第二行有n个字母(‘R’或’B’)，每个第i个字母表示PH试纸第i段的颜色。
接下来m行，第i行有一个大写字母 ci(‘R’或’B’)和一个正整数qi ，用空格隔开，表示查询颜色ci 第qi 次出现的位置。

输出描述:

共m行，第i行一个整数，表示查询结果，若颜色ci出现次数少于qi次，则输出-1，否则输出颜色qi第ci次出现的位置。

示例1

输入

复制

2 2
RB
R 1
B 1

输出

复制

1
2

示例2

输入

复制

2 2
BB
R 1
B 2

输出

复制

-1
2

示例3

输入

复制

3 3
BRB
B 1
B 2
R 1

输出

复制

1
3
2

备注:

对于100%的测试数据：
1 ≤ n, m ≤ 1000000
所有输入数据不超过1000000。
数据量较大，注意使用更快的输入输出方式。

解题报告：

  直接维护两个前缀和，复杂度O(m)。

AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<set>
#include<string>
#include<cmath>
#include<cstring>
#define ll long long
#define pb push_back
#define pm make_pair
using namespace std;
const int MAX = 2e6 + 5;
int n,m;
char s[MAX];
int R[MAX],B[MAX],r,b;
int main()
{
	cin>>n>>m;
	scanf("%s",s+1);
	getchar();
    memset(R,-1,sizeof R);
    memset(B,-1,sizeof B);
	for(int i = 1; i<=n; i++) {
		if(s[i] == 'R') R[++r] = i;
		if(s[i] == 'B') B[++b] = i;
	}
	char ch[4];
	int x;
	while(m--) {
		scanf("%s%d",ch,&x);
		if(ch[0] == 'R') printf("%d\n",R[x]);
		else printf("%d\n",B[x]);
	}
	return 0 ;
}

进一步扩展：如果是1e5种颜色呢？  答：那也很简单啊1e5个vector就完事了。 

再一步扩展：如果两种颜色带修改呢？  答：那就建两棵线段树，对应位置如果是当前线段树维护的颜色， 那就置为1 ，然后查询第k大。

终极扩展：那如果是1e5种颜色带修改呢？  答：那就主席树。每种颜色都建一棵4*n这么大的线段树，因为同构 所以压缩到一棵主席树里，查询就找该颜色对应的线段树 在线段树上二分，线段树上维护的是 是该树维护的颜色则该点记为1 其他颜色记为0。（%%%syt）