蓝桥杯练习3——分糖果

问题描述

最近暑期特训算法班的同学们表现出色，他们的老师肖恩决定给他们分发糖果。肖恩购买了 nn 个不同种类的糖果，用小写的阿拉伯字母表示。每个糖果必须分发给一个同学，并且每个同学至少要分到一个糖果。同学们的开心程度定义为他们所分到的糖果组成的字符串 s[i]s[i] 的字典序。肖恩希望同学们的开心程度相差尽量小，因此他要找到一种方案，使得所有糖果组成的字符串中字典序最大的字符串尽可能小。请输出能够实现字典序最小可能的 max⁡(s[1],s[2],s[3],...,s[x])max(s[1],s[2],s[3],...,s[x]) 。

输入描述

第一行输入两个整数 nn 和 xx ，分别表示有 nn 个糖果 xx 个同学。

第二行输入一个长度为 nn 的字符串 SS ， S[i]S[i] 表示第 ii 个糖果的种类。

数据保证 1≤n≤106,1≤x≤n,S[i]∈[′a′,′z′]1≤n≤106,1≤x≤n,S[i]∈[′a′,′z′] 。

输出描述

输出一个字符串，为所有糖果组成的字符串中字典序最大的字符串最小的可能值。

样例输入:

6  2

caabdc

样例输出:

abccd

说明

一个最优分配方案是一个同学拿到 abccdabccd ，一个同学拿到 aa 。

思路：首先我们需要明白S序列可以被分为x段任意的长度及组合，其次解决判断分成的字符串大小问题；例如当字符串aba和ba比较时，因为b字符顺序大于a，所以ba>aba,而当字符串aba和ab判断时，因为前两个字母相同，aba因为是三个字符>两个字符因此aba>ab;因此将我们刚刚分类的x个字符串记为字典序S(x)，当最大的字典序S（x）为最小序列时，各糖果之间的差距最小，即求min[max S(x)]。

举例：

import os
import sys
def ii():
    return int(read())
def il():
    return list(map(int,read().split()))
def read():
    return sys.stdin.readline().strip()
t = 1#只有一个测试用例
ans = []#用于存储答案
for _ in range(t):#循环处理这一个测试用例
    n , x = il()#从标准输入读取两个整数n,x
    s = read()#读入字符串
    s = sorted(s)#对字符串进行排序
    if s[0] != s[x-1]:
    #判断最小字符数量小于x
        print(s[x-1])
    #排序后字符串第一个字符和第x-1个字符不相同，则输出第x-1个字符
    sys.exit(0)
  #最小字符数量大于等于x
    ans.append(s[0])
    #如果最小字符的数量小于或等于 x，则将最小的字符添加到答案列表中。然后检查第 x 个字符和最后一个字符是否相同。
    # 如果不同，则将 x 到最后一个字符的所有字符添加到答案列表中；如果相同，则将整个字符串的剩余部分均分并添加到答案列表中。
    if s[x] != s[n-1]:
    #说明剩下的不全是一个字符，我们直接把【x,n-1】所有字符填到ans中
        for i in range(x,n):
            ans.append(s[i])
    else:
    #剩下的全是一个字符，直接均分
        for i in range((n-1)//x):
            ans.append(s[n-1])
    print(''.join(ans))