给定一个长度为n的char型数组,由小写字母组成,然后要求q次查询,每次查询给出三个值l、r、k,输出l至r之间第k小的字母,如果k越界则输出“Out of range”。
条件限制:1<=n<=5*10^4,1<=n<=10^5,1<=l<=r<=n,1<=k<=n.
输入:第一行n,q,空格隔开;第二行n个小写字母,空格隔开;剩下q行l,r,k,空格隔开。
输出:有第k小的字母就输出,没有就输出“Out of range”。
(原题目)
首先我的思路是将数组中l到r的值赋给一个新的数组,利用sort排序后输出第k个字母,但这样只有两个案例通过,其余超时;
然后我又改进思路,加入getchar_unlocked()输入,利用一个长度为26的辅助数组b统计l到r之间出现的字母数字,让k依次-=b[i++],k<=0时将‘a’+i输出为所求,但是依然有一个案例超时,且内存最大是1500多KiB,效果不理想;
于是我又改进思路,直接将字母出现次数统计在a[n][26]的数组中,输入完毕对a数组依次累加,需要求l到r之间的字母出现次数时,只需要建一个辅助数组b[26],令i从0至25,b[i]=a[r][I]-a[l-1][I]即可,这时再让k依次-=b[i],当k<=0时输出'a'+i即为所求。这次时间将累加过程放在开始,不用每次查询都要累加,节省了大量时间。
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int a[100000][26]={0};
inline void input(int *n){
register char c=getchar_unlocked();
*n=0;
for(;(c<'0'||c>'9');c=getchar_unlocked());
for(;(c<='9'&&c>='0');c=getchar_unlocked()){
*n=*n*10+c-48;
}
}
int main(){
int n,q,i=1,j,b[26];
input(&n);
input(&q);
int l,r,k;
register char c=getchar_unlocked();
for(;(c<'a'||c>'z');c=getchar_unlocked());
for(;(c<='z'&&c>='a');c=getchar_unlocked()){
a[i++][c-'a']++;
}
for(i=2;i<=n;i++){
for(j=0;j<26;j++){
a[i][j]+=a[i-1][j];
}
}
while(q--){
input(&l);
input(&r);
input(&k);
if(k>(r-l+1)){
printf("Out of range\n");
}else{
for(i=0;i<26;i++){
b[i]=a[r][i]-a[l-1][i];
k-=b[i];
if(k<=0)break;
}
printf("%c\n",'a'+i);
}
}
}
提交的结果: