数组中选取红蓝数字，计算乘积之和 - Python3 实现

数组中选取红蓝数字，计算乘积之和 - Python3 实现

问题描述: 小红拿到一个数组，其中一些数字被染成红色，另一些染成蓝色。小红准备从中取2个数，选出的两个数的乘积为选择的方案权值，选择一红一蓝才是合法的。

输入格式: 第一行输入一个正整数 n,代表数组大小。 第二行输入 n 个正整数 ai，代表小红拿到的数组。 第三行输入一个由'R' 和 'B' 组成的长度为 n 的字符串，第 i 个字符是 'R' 表示第 i 个元素被染成红色。

输出格式: 请输出所有合法的选数方案的权值之和，并对 10 的九次方 + 7 取模。

思路: 将红色和蓝色的数分别存入两个数组中，然后分别排序，从小到大枚举红色数组和蓝色数组的组合即可，注意判断两个数是否颜色不同。

Python3 代码:

MOD = 10**9 + 7
n = int(input())
a = list(map(int, input().split()))
colors = input()

red = []
blue = []
for i in range(n):
    if colors[i] == 'R':
        red.append(a[i])
    else:
        blue.append(a[i])

red.sort()
blue.sort()

sum = 0
for i in range(len(red)):
    for j in range(len(blue)):
        sum = (sum + (red[i] * blue[j]) % MOD) % MOD

print(sum)

代码解释:

1. 定义 MOD 为 10 的九次方 + 7。
2. 输入数组大小 n 和数组元素 ai。
3. 输入颜色字符串 colors。
4. 创建两个空列表 red 和 blue，分别存储红色和蓝色数字。
5. 遍历数组，根据颜色将数字添加到相应的列表中。
6. 对 red 和 blue 列表进行排序。
7. 初始化 sum 为 0，用于存储所有合法方案的乘积之和。
8. 使用两层循环枚举 red 和 blue 中的数字组合。
9. 计算每个组合的乘积，并对 MOD 取模，累加到 sum 中。
10. 输出 sum。

注意:

• 代码中的注释可以帮助你理解代码的逻辑。
• 代码中的 % MOD 操作是为了确保结果在模 10 的九次方 + 7 的范围内。
• 你可以根据实际情况修改代码，例如改变数组大小、输入格式等。