字符串拼接-CSDN博客

本文链接：https://blog.csdn.net/weixin_48701352/article/details/142123250

题目描述

给定 M（0 < M ≤ 30）个字符（a-z），从中取出任意字符（每个字符只能用一次）拼接成长度为 N（0 < N ≤ 5）的字符串，

要求相同的字符不能相邻，计算出给定的字符列表能拼接出多少种满足条件的字符串，

输入非法或者无法拼接出满足条件的字符串则返回0。

输入描述

给定的字符列表和结果字符串长度，中间使用空格(" ")拼接

输出描述
满足条件的字符串个数

例1

输入
abc 1
输出
3
说明
给定的字符为a,b,c，结果字符串长度为1，可以拼接成a,b,c，共3种

例2

输入
abcde 5
输出
120
说明
全排列

Python实现

from collections import Counter

def get_ways(strs,n):
    if not strs.isalpha() or not strs.islower or n < 1 or n > 5 or len(strs) < n:
        return 0
    
    counter = Counter(strs) #字典计数，统计每个字母有多少个
    
    return process('',n,counter)
    
def process(pre_char,rest,counter):
    if rest == 0:
        return 1
    else:
        total = 0
        for char in counter:
            if char != pre_char and counter[char] > 0:
                counter[char] -= 1
                total += process(char,rest-1,counter)
                counter[char] += 1 #回溯
        return total
    

strs,n = input().split()
n = int(n)
get_ways(strs,n)