您的位置:首页 > 其它

HDU 5431

2015-09-08 10:52 369 查看
由于最长不超过30个字符(由K的范围确定),于是,枚举所有的字符串,二分中使用二分就可以确定第K小了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

char str[20010];
int bits[32];
int save[32][20010];
int tmp[32];

int findindex(int i,int key){
int l=1,r=save[i][0];
int m,ans=0;
while(l<=r){
int m=(l+r)/2;
if(save[i][m]<=key){
ans=m;
l=m+1;
}
else r=m-1;
}
return ans;
}

int main(){
int kase,now,T,K;
scanf("%d",&kase);
for(int i=1;i<32;i++) bits[i]=(1<<i);
while(kase--){
scanf("%s",str);
int len=strlen(str);
for(int i=0;i<31;i++) save[i][0]=0;
for(int i=0;i<len;i++){
now=0;
for(int j=i;j<=min(i+29,len-1);j++){
now=(now<<1)+(str[j]-'A');
save[j-i+1][++save[j-i+1][0]]=now;
}
}
for(int i=1;i<=30;i++) sort(save[i]+1,save[i]+save[i][0]+1);
for(int i=1;i<=30;i++) {
save[i][0]=unique(save[i]+1,save[i]+save[i][0]+1)-(save[i]+1);
//	cout<<save[i][0]<<endl;
}
//	cout<<(char)(save[1][1]+'A')<<endl;
scanf("%d",&T); int ans,ansbit;
while(T--){
scanf("%d",&K);
ans=-1;
for(int i=1;i<=30;i++){
//		cout<<K<<endl;
if(K>bits[i]-save[i][0]){
K-=(bits[i]-save[i][0]);
continue;
}
int l=0,r=bits[i]-1;
while(l<=r){
int m=(l+r)>>1;
int pos=findindex(i,m);
//		cout<<m<<" "<<pos<<endl;
if(K>m+1-pos) l=m+1;
else if(K<m+1-pos) r=m-1;
else{
ans=m;
while(ans==save[i][pos]&&pos!=0){
ans--; pos--;
}
break;
}
}
//		system("pause");
if(ans!=-1){
ansbit=i;
break;
}
//		else K-=(bits[i]-save[i][0]);
}
int cnt=0;
//	cout<<ans<<endl;
while(ansbit--){
tmp[cnt++]=ans%2;
ans/=2;
}
while(cnt--)
printf("%c",tmp[cnt]+'A');
puts("");

}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: