题干:
链接: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)