程序设计——week15实验课限时训练（瑞瑞爱上字符串））

瑞瑞爱上字符串

题目描述

瑞瑞最近迷上了字符串，因此决定出一个字符串的题。给定两个正整数 N、K，考虑所有由 N - 2 个 a 和 2 个 b 组成的字符串，要求输出其中字典序第 K 小的。
例如当 N = 5 时，共有如下 10 种组成方式：

    1.aaabb
    2.aabab
    3.aabba
    4.abaab
    5.ababa
    6.abbaa
    7.baaab
    8.baaba
    9.babaa
    10.bbaaa

Input：
多组数据，第一行给定 T，表示数据组数。（1 ≤ T ≤ 1e4）
对于每组数据，给出两个正整数 N、K。（3 ≤ N ≤ 1e5, 1 ≤ K ≤ min(2e9, N * (N-1) / 2 )）
N 的总和不会超过 1e5。
Output：
对于每组数据，输出长度为 N 的字典序第 K 小的字符串。
Example：

    Input：
    7
    5 1
    5 2
    5 8
    5 10
    3 1
    3 2
    20 100

    Output：
    aaabb
    aabab
    baaba
    bbaaa
    abb
    bab
    aaaaabaaaaabaaaaaaaa

思路分析

这道题我由于细节没处理好，直到时间的最后关头都还WA着…
首先，这道题的突破点是寻找b适宜的位置，可以根据题面中N=5情况下那10种情况进行分析。

根据表中可以寻找规律如下：
令字符串有n个字符，令第一个b的位置为i，那么第二个b的位置就为n、n-1、n-2…i+1。所以第一个b的位置为i的情况就有n-i 种。比如在上表中，第一个b的位置为4、3、2、1时，第二个b的位置依次对应有1、2、3、4种情况。
所以只需要根据输入的k值先确定出第一个b的位置，再根据第一个b的位置和k值综合考虑确定出第二个b的位置。

(1).
int rnk=1,level=1;
for(int i=2;rnk<k;i++){//累加当前名次数rnk
	rnk+=i;level=i;//level代表(n-第一个b的位置)
}
//当前名次数如果大于k，就说明所求字符串中第一个b的位置就是n-level,之后判断第二个b的位置，进行第（2）步。
(2).
while(rnk>k){
	r++;rnk--;
}

代码实现

#include<cstdio>
#include<algorithm>
using namespace std;

int t;
int main(){
	scanf("%d",&t);
	int n,k;
	while(t--){
		scanf("%d%d",&n,&k);
		int rnk=1,level=1;
		for(int i=2;rnk<k;i++){
			rnk+=i;
			level=i;
		}
		//printf("rnk:%d  level:%d\n",rnk,level);
		int l=n-level,r=n-level+1;//n-level表示第一个b的位置，n-level+1表示
		while(rnk>k){
			r++;
			rnk--;
		}
		//printf("l:%d r:%d\n",l,r);
		for(int i=1;i<=n;i++){
			if(i==l||i==r) printf("b");
			else printf("a");
		}
		printf("\n");
	}
	return 0;
}