Python解题 - CSDN周赛第29期 - 争抢糖豆

本期问哥是志在必得，这本算法书我已经觊觎许久，而之前两次因为种种原因未能如愿。因此，问哥这几天花了不少时间，把所有之前在每日一练做过的题目重新梳理了一遍。苦心人，天不负，感谢官方大大！

---

第一题：订班服

小A班级订班服了！ 可是小A是个小糊涂鬼，整错了好多人的衣服的大小。 小A只能自己掏钱包来补钱了。 小A想知道自己至少需要买多少件衣服。

输入描述：第一行输入一个整数n。(1<=n<=100)表示衣服的数量。 以下n行输入n个尺码。表示订单中衣服的尺码。 接下来n行输入n个尺码。小A订的衣服尺码。 尺码表：M,S,L,XL,XLL,XLLL,XLLLL,XLLLLL。

输出描述：输出至少需要买多少件衣服。

示例：

	示例
输入	2 XL X M X
输出	1

分析

简单题。需要 n 件衣服，买了 n 件衣服，所以只要把需要的 n 件衣服每种尺码的个数，减去已购买的衣服相应尺码的个数，如果大于0就说明该尺码还需要购买。所以最直接的办法就是用哈希表记录每种尺码的个数，然后再逐个检查。

Python已经提供了内置Counter类，可以自动生成字典，而且支持加减法，连逐个检查这一步都省去了：直接相减，剩下的数字加在一起就是答案。

参考代码

n = int(input().strip())
arr1 = [input().strip() for _ in range(n)]
arr2 = [input().strip() for _ in range(n)]
from collections import Counter
clothes = Counter(arr1) - Counter(arr2)
print(sum(clothes.values()))

---

第二题：争抢糖豆

抓糖豆，小Q与小K都喜欢吃糖豆。 但是糖豆分两种，超甜糖豆和普通糖豆。 现在有w个超甜糖豆和b个普通糖豆。 小Q和小K开始吃糖豆，他们决定谁先吃到超甜糖豆谁就获胜。 小K每次吃的时候会捏碎一颗糖豆。 小Q先吃，小Q想知道自己获胜的概率。 如果两个人都吃不到超甜糖豆小K获胜。

输入描述：输入两个整数w,b。(0<=w,b<=1000)

输出描述：答案保留9位小数。

示例：

	示例
输入	1 3
输出	0.500000000

分析

也是以前考过的老题了。可以用递归或动态规划来做，但本题的状态转移不太容易一眼发现，所以可能不少人会觉得难。因为问到概率（胜率），所以本质上还是需要用数学来表达。

以动态规划为例（递归容易超时），我们用  表示当有  颗超甜糖豆，和  颗普通糖豆时自己的胜率。因为先吃到超甜糖豆就获胜了，所以自己要想获胜，只能分成两种情况：

1. 先吃到超甜糖豆，概率是   ，此情况下直接获胜；
2. 先吃到普通糖豆，但是对手也吃到普通糖豆，所以游戏继续，自己还有获胜的可能。（这里有一个特判的情况：如果只有一颗普通糖豆，而自己先吃到普通糖豆的话，无论如何也是输，后面自然就不用算了。）因此，自己和对手都吃到普通糖豆的概率是   。（如果一时看不懂可以多琢磨几遍，乘号左边是自己吃到普通糖豆的概率，右边是自己吃完后对方也吃到普通糖豆的概率，看懂了再继续。）

如果没有“捏碎糖豆”的操作，分析到这就结束了，状态转移就是把这两种情况的胜率加在一起，方程如下：

（因为自己和对手总共吃了两颗普通糖豆，所以上面第二种情况的概率还要乘以  才是胜率。）

如果上面的内容理解了，我们再来分析“捏碎糖豆”的情况。

捏碎糖豆也有两种情况：

1. 捏碎了普通糖豆。影响不大，但是上面的第二种状态要接着乘上捏碎普通糖豆的概率，再乘以  。合在一起的胜率就是  。
2. 捏碎了超甜糖豆。则二人虽不分胜负，理论上自己还存在胜利的可能（如果还剩下超甜糖豆的话），但是同样地，状态转移方程变了，胜率变成了： 。

这两种捏碎糖豆的情况属于同一决策层级，可以加在一起。于是把捏碎糖豆考虑进来，得到最终的状态转移方程如下：

此外，如之前所述，还要考虑几个特判的情况：

1. 没有超甜糖豆，胜率为0，不用计算。
2. 没有普通糖豆，胜率100%。
3. 只有一颗普通糖豆，胜率为 ，因为如果自己先吃到普通糖豆，必输。

很显然，上面第二、三可以合并，而第一条可以在初始化的时候把  的时候设置为0。代码如下：

参考代码

w, b = map(int, input().split())
dp = [[0]*(b+1) for _ in range(w+1)]
for i in range(1, w+1):
    for j in range(b+1):
        if j <= 1:
            dp[i][j]=i/(i+j)
        else:
            dp[i][j]=i/(i+j)+j/(i+j)*(j-1)/(i+j-1)*((j-2)/(i+j-2)*dp[i][j-3]+i/(i+j-2)*dp[i-1][j-2])
print(f"{dp[w][b]:.9f}")

输出结果的时候要注意，题目要求必须保留9位小数，空位用0补全，所以要设置占位符。

---

第三题：走楼梯

现在有一截楼梯，根据你的腿长，你一次能走 1 级或 2 级楼梯，已知你要走 n 级楼梯才能走到你的目的楼层，请实现一个方法，计算你走到目的楼层的方案数。

输入描述：输入整数n。(1<=n<=50)

输出描述：输出方案数。

示例：

	示例
输入	5
输出	8

分析

很明显，答案是斐波那契数列。因为一次只能走 1 级或 2 级楼梯，所以第 n 级阶梯可以由 n-1 级阶梯走过来，也可以由 n-2 级阶梯走过来。如果用函数  表示走上第 n 级阶梯的方案数，可得

因为本题 n 范围较小（1<=n<=50），所以使用递归来做应该也没问题。不过更常用的做法是递推，也算是斐波那契的模板题了。

参考代码

n = int(input().strip())
a = b = 1
for _ in range(n):
    a, b = b, a+b
print(a)

---

第四题：打家劫舍

一个小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。 给定一个代表每个房屋存放金额的非负整数数组，计算不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。

输入描述：输入一个正整数n代表房屋的数量(n≤100)，接着输入n个非负整数代表每间房屋的现金数量

输出描述：小偷能偷取的最大金额。

示例：

	示例
输入	4 1 2 3 1
输出	4

分析

力扣原题，经典的打家劫舍系列，但凡刷过点题的相信都做过。而且这里选取的是该系列最简单的一道，动态规划入门题，相关题解太多了，这里问哥只简单说两句吧。

因为相邻的房屋不能同时被盗，所以小偷在当前房屋只有两种选择：不偷当前房屋——继承上个房屋可偷取的的最大金额，偷当前房屋——上上个房屋可偷取的最大金额（因为上个房屋不能偷）加上当前房屋的金额，而当前房屋可偷取的最大金额就等于这两种选择中较大的金额。

如果用 代表当前房屋可偷取的最大金额， 表示当前房屋的金额，可用公式表示如下：

类似斐波那契数列，可以看出  仅由  和  得到（ 是给定数组），所以可以使用滚动数组优化空间，换句话说就是只需要额外两个变量循环保存  和  的值即可。

参考代码

n = int(input().strip())
arr = [int(item) for item in input().strip().split()]
a = b = 0
for i in arr:
    a, b = b, max(b, a+i)
print(b)