CF1348 C. Phoenix and Distribution

CF1348 C. Phoenix and Distribution

Question

给你n个字符，要求你将其分为k份，每份个数至少有1个，使得新组合的字符串中字典序最大的字符串的字典序尽可能小，打印这个字符串。

Solution

分类讨论
字典序最小，先排序。

1. $s_1\neq s_k$，答案为 $s_k$，此时剩余部分直接放在 $s_1$后面所得解最优。
2. $s_1=s_k \ and \ s_{k+1}=s_{n}$，答案为 $s_1+\lceil\frac{(n-k)}{k}\rceil\times s_{n}$
3. $s_1=s_k \ and \ s_{k+1}\neq s_{n}$，答案为 $\sum _{i=k}^{n}s_i$

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 1e9 + 7;
const ll  maxn = 1e6 + 5;
const int N = 1e5 + 5;

void solve(){
	string s;
	int n,k;
	cin>>n>>k>>s;
	sort(s.begin(),s.end());
	if(s[0]!=s[k-1]){
		cout<<s[k-1]<<'\n';
	}else{
		cout<<s[0];
		if(s[k]!=s[n-1]){
			for(int i=k;i<n;i++) cout<<s[i];
		}else{
			for(int i=0;i<(n-1)/k;i++) cout<<s[n-1];
		}
		cout<<'\n';
	}
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;cin>>T;
	while(T--){
		solve();
	}
	return 0;
}