华为OD机试-字符串分割

题目描述

给定一个非空字符串S，其被N个‘-’分隔成N+1的子串，给定正整数K，要求除第一个子串外，其余的子串每K个字符组成新的子串，并用‘-’分隔。对于新组成的每一个子串，如果它含有的小写字母比大写字母多，则将这个子串的所有大写字母转换为小写字母；反之，如果它含有的大写字母比小写字母多，则将这个子串的所有小写字母转换为大写字母；大小写字母的数量相等时，不做转换。
输入描述:
输入为两行，第一行为参数K，第二行为字符串S。
输出描述:
输出转换后的字符串。
示例1
输入
3
12abc-abCABc-4aB@
输出
12abc-abc-ABC-4aB-@
说明
子串为12abc、abCABc、4aB@，第一个子串保留，后面的子串每3个字符一组为abC、ABc、4aB、@，abC中小写字母较多，转换为abc，ABc中大写字母较多，转换为ABC，4aB中大小写字母都为1个，不做转换，@中没有字母，连起来即12abc-abc-ABC-4aB-@
示例2
输入
12
12abc-abCABc-4aB@
输出
12abc-abCABc4aB@
说明
子串为12abc、abCABc、4aB@，第一个子串保留，后面的子串每12个字符一组为abCABc4aB@，这个子串中大小写字母都为4个，不做转换，连起来即12abc-abCABc4aB@

代码实现

# coding:utf-8
class Solution:
    def stringSplit(self, k, s):
        begin = 0
        end = 0
        # 遍历字符串，遇到第一个分隔符的时候跳出，将第一个子串存下来，不需要做大小写转换
        for i in s:
            if i == '-':
                break
            else:
                end += 1
        res = s[begin:end]
        begin = end + 1

        # 从第二个子串开始判断
        while begin < len(s):
            tmp = ''
            lower = 0
            linefix = 0
            # 每K个字符串作为一个新的子串长度
            for i in range(k):
                while begin + i + linefix <= len(s) - 1 and s[begin + i + linefix] == '-':  # 遇到分隔符，直接跳过不处理，分隔符计数器+1
                    linefix += 1
                if begin + i + linefix > len(s) - 1:  # 超出原字符串长度，跳出
                    break
                if s[begin + i + linefix].islower():  # 当前字符为小写，则小写计数器加1
                    lower += 1
                elif s[begin + i + linefix].isupper():  # 当前字符为大写，则小写计数器减1
                    lower -= 1
                tmp = tmp + s[begin + i + linefix]

            # 小写字符多则全部转小写，反之，转大写
            if lower > 0:
                res = res + '-' + tmp.lower()
            if lower < 0:
                res = res + '-' + tmp.upper()
            if lower == 0:
                res = res + '-' + tmp
            begin = begin + k + linefix
        return res


if __name__ == '__main__':
    k = input("K:")
    s = input("s:")
    solution = Solution()
    print(solution.stringSplit(int(k), s))