交换前缀 CodeForces 152 C.Pocket Book

强殖装甲凯普

于 2021-09-21 15:37:38 发布

阅读量136

点赞数

分类专栏：算法题文章标签： python 算法

本文链接：https://blog.csdn.net/qq_38163755/article/details/120402070

版权

算法题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

文章目录

问题
解决

问题

现在有n个长度为m的字符串，编号为从1到n，每个字符串由m个大写字母组成。现在你可以完成以下操作，请你任选第 i 个字符串和第 j 个字符串，并交换长度为 k 的前缀。你可以在变换后的基础上进行任意次这样的操作。

例如：ABCD 和 EFGH，令k=2则变为 EFCD 和 ABGH。

此时对于新的字符串 EFCD 和 ABGH 令k=1则变为 AFCD 和 EBGH。

显然变化后的字符串是不同的。

现在请问你可以生成多少个不同的字符串。包含原串本身。
在这里插入图片描述
样例输入：
2 3
ABC
DEF
样例输出
8

解决

由于是交换前缀，其实意味着是相同位置的字母交换，可以看做计算构成字符种类的排列组合，m长的字符串，每一位可用字符数是n个字符串在该位上不同字母的个数，然后乘起来即可。

python3实现：

tmp = list(map(int, input().split()))
n, m = tmp[0], tmp[1]
matrix = []
for i in range(n):
    tmp = input()
    matrix.append(tmp)
res = 1
for j in range(m):
    tmp = set()
    for i in range(n):
        tmp.add(matrix[i][j])
    res *= len(tmp)
    res = res % 1000000007
print(res)