Each person walks at his/her fixed speed. If two people cross the bridge together, they must walk at the pace of the slower one. How fast can you get all N people over the bridge?
经典的过桥的问题,第一眼很容易觉得这会是贪心解法的题目。因为每次都要有人过桥回来送手电筒,那么我们每次让跑的最快的送就可以。但是考虑例子v = [1,2,5,10],如果我们每次让v[0]跑来会送手电筒,那么总时间为2 + 1 + 5 + 1 + 10 = 19,但实际上最快的时间是17(当然这道题贪心也可以做就是没那么直观,感兴趣的可以google):
- v[0]和v[1]过桥,用时2
- v[0]带手电筒回,用时1
- v[2]和v[3]过桥,用时10
- v[1]回,用时2
- v[0]和v[1]过桥,用时2
我们可以看出贪心之所以不行是因为每次跑的最快的人都要来回过桥,从而我们浪费了让两个相对慢的人一起过桥缩短总时间的机会。
如果我们从子问题开始考虑用DP来解决这个问题,我们把所有人按照用时从少到多sort,得到数组v。我们考虑前i个人所用的总时间DP[i],我们有递推公式:
- DP[i] = DP[i - 1] + v[0] + v[i],前i - 1个人已经过桥,最快的人回来送电筒然后和第i个人一起过桥
- DP[i] = DP[i - 2] + v[0] + v[i] + 2 * v[1],前i - 2个人过桥,最快的人送电筒回来,第i - 1和第i个人过桥,第二快的人送电筒回来然后和最快的一起过桥
我们根据这个递推公式计算即可。时间复杂度空间复杂度均为O(n),用滚动数组可以优化到常数空间,普通DP代码如下:
优化后的代码:
No comments:
Post a Comment