区间动态规划(记忆化搜索 @ Python) - 石头合并 粗浅理解

'''
记忆化搜索,分治
P1880 [NOI1995]石子合并 @ LuoGu
https://www.luogu.org/problemnew/show/P1880
题目描述
在一个**圆形操场**的四周摆放N堆石子,现要将石子有次序地合并成一堆.规定每次只能选**相邻**的2堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。

试设计出1个算法,计算出将N堆石子合并成1堆的最小得分和最大得分.

输入输出格式
输入格式:
数据的第1行试正整数N,1≤N≤100,表示有N堆石子.第2行有N个数,分别表示每堆石子的个数.

输出格式:
输出共2行,第1行为最小得分,第2行为最大得分.

输入输出样例
输入样例#1: 
4

输出样例#1: 
43
54
'''

可能对新人(OIer && ACMer请绕道)有用的新概念

前缀和
对应一个 List[int] 维护一个 角标 i 存着这个 List[int] 0 ~ i 的数的和,方便通过O(1)减法得到区间和。

因为是圆形操场,输入为 [4,5,9,4] 的话,存在把 [4,4] 合并的情况,学过离散数学的同学应该看过圆桌问题(一个桌子5个人,能坐多少种不同的顺序),这就需要引入一个大一点的序列以便处理环现象。

4 5 9 4 -> 5 9 4 4 -> 9 4 4 5 -> 4 4 5 9 -> (4, 5, 9, 4)
那么简化一下,可以得到
石头:    [(0), 4, 9, 5, 4, 4, 9, 5, (4)],和
前缀和:[0, 4, 13, 18, 22, 26, 35, 40, 44]

你可以想象一个长为4的窗口,每移动一位,就能复现上面环的各种情况。括号里面的0 和 4 加进来用于处理边界情况,更方便利用前缀和来获得区间和

记忆化搜索有的地方也叫分治法,在国外这手段也被归并到动态规划。不过因为涉及递归,所以比一般的顺序DP会慢一点。
上代码,这里只做min的情况,max的话基本上就是这个的映射版:

class Solution(object):
    def getSum(self,i,j):
        return self.s[j] - self.s[i-1]
    def mergeStone(self,nums):
        '''
        nums : List[int]
        '''
        n = len(nums)
        s = []
        nums = [0] + nums + nums
        L = 1
        R = 2*n
        s = [i for i in nums]
        for i in range(1,len(s)):
            # 前缀和
            s[i] = s[i-1] + s[i]
        self.s = s
        # extra 1 for lower limit. 
        self.min_memo = [[None for b in range(n+n+1)] for a in range(n+n+1)]
        
        self.dp_min(L,R)
        minres = float('inf')
        for i in range(1,n+1):
           minres = min(minres,self.min_memo[i][i+n-1])
        print(minres)
    def dp_min(self,i,j):
        '''
        i : 区间始
        j : 区间末
        s : 前缀和,用于访问
        '''

        if j==i:
            self.min_memo[i][j] = 0
            return 0
        elif self.min_memo[i][j] != None:
            return self.min_memo[i][j]
        else:
            this_cost = self.getSum(i,j)
            tres = float('inf')
            for c in range(i,j):
                tres = min(tres,self.dp_min(i,c)+self.dp_min(c+1,j)+this_cost)
            self.min_memo[i][j] = tres
            return tres

区间DP的思想

其实很简单
通过递归来把大区间[i,j], 以 i <= c self.dp_min(i,c),self.dp_min(c+1,j) 为两个子区间的最小代价。
当两个子区间带着各自的代价形成,最后形成 [i,j],无论两个子区间是怎样的,根据定义,这一步合成的代价一定就是[i,j] 的石子总和,可以通过前缀和轻松求出。

BASE CONDITION
self.min_memo[i][i] = 0, 只能代表当前单独第 i 堆,无任何合并操作,所以不会产生任何代价。
self.min_memo[i][i+1] 则是两堆石子合并,其实最终也就通过前缀和来求。

大佬巨佬们多多指教小弟做人,我写这个主要想把所思所想记录下来,有不少新概念和操作都是看过大神们提一提才慢慢懂。这篇算是小白文吧,毕竟我想通过或者CSDN搜,大佬们都写得像是准备给水平已经很不错的读者。

你可能感兴趣的:(区间动态规划(记忆化搜索 @ Python) - 石头合并 粗浅理解)