POJ 1416 - Shredding Company + Python(DFS*)

博客详细介绍了POJ 1416问题的解决方法,通过深度优先搜索(DFS)策略进行字符串切分。重点在于递归思路的阐述,强调在搜索过程中如何处理特殊情况,包括不回溯的特殊情况以及在找到满足条件的方案时如何保存和更新最优方案。此外,还提醒了在递归过程中避免直接修改队列的值以防止错误影响。
摘要由CSDN通过智能技术生成

(不一样的递归思想,值得再看。)

题目链接:

# https://exp-blog.com/algorithm/poj/poj1416-shredding-company/
# https://blog.csdn.net/lyy289065406/article/details/6647969

一 特别注意

1. 关于递归思路,参考别人的设计思想:

        以“12346”为例,当进行深度搜索时:

        第一步,先切下一个“1”,然后递归去切“2346”;
        第二步,再切下一个“12”,然后递归去切“346”;
        第三步,再切下一个“123”,然后递归去切“46”;
        第四步,再切下一个“1234” 然后递归去切“6”
        第五步,再切下“12346”。

PS:我没有进行第五步拆分,而是在初始阶段对第五步这种特殊情况进行了判断处理;这样的话,可以保证每次拆分字符串都可以拆成前后两段:

## 循环
        E = len(Data)
        store = []
        for i in range(E-1): # 拆成前后两个字符串
            store.append([Data[:(i+1)],Data[i+1:]])

2. 特别地,当遇到满足条件的方案后,要考虑特殊处理:

2.1 将当前方案加入到列表中,继续分析:

        2.2.1 当前方案中节点长度大于1时,算法不进行回溯,而是继续执行,寻找新的可行方案

        2.2.2 当前方案中节点的长度为1时,算法进行回溯;回溯之后,弹出当前的节点,再继续寻找新的方案,而不是return至上一层。

        2.2.3 特别注意:在执行上述过程时,要修改相应的参数信息。(具体见dfs子程序)

2.2 另外,递归过程不要直接更新队列的值,也不要将队列直接赋值给新变量。否则,队列一经修改,其他变量也会更改。

3 在保存方案时:(体现在第二版中)

3.1 如果当前方案的和值更大,则清空原有方案,再存储当前方案

3.1 如果当前方案的和值等于之前方案的和值,那么在存储当前方案之前不必清空原有方案。

二 代码实现

版本二:20211226

# 20211226 重做
# http://poj.org/problem?id=1416
import collections

class Solutions:
    def DFS(self):
        self.TARGET = 0
        self.ways = [] # 用于存放方案
        self.Sum_count = collections.defaultdict(list) #保存当前部分和所出现的次数
        LEN = len(subData)
        store = []
        ## 1 将数据划分为两个子串,对后一个子串进行DFS搜索
        for i in range(LEN-1):
            store.append([subData[:i+1],subData[i+1:]])
        LEN = len(store)
        ## 对每一组子串进行分析处理
        for i in range(LEN):
            # 取出待处理的数据
            sub1 = store[i][0]
            sub2 = store[i][1]
            # 处理:
            sum = 0 + int(sub1)# 部分和sum
            # 判断部分和是否超过了目标
            if sum < int(target):
                queue = [sub1]
                self.dfs(sub2,sum,queue)
        ## 对所有方案进行分析

        return self.ways

    def dfs(self,date,sum,queue):
        ## 2 首先判断是否满足目标要求,满足则保存方案;然后,进一步判断date的长度是否为1,为1则返回,否则继续拆分成子串+DFS搜索
        if sum + int(date) <= int(target):
            # 判断其部分和是否更大
            if sum + int(date) > self.TARGET: # 1 当大于时,获得更大值,在存储之前清空一下
                self.TARGET = sum + int(date)
                # 清空原有方案和字典
                self.ways = []
                self.Sum_count = collections.defaultdict(list)  # 保存当前部分和所出现的次数
                # 保存当前方案,注意:保证顺序不要乱
                new_queue = []
                new_queue.append(date)  # 先添加到队头
                for qi in queue:
                    new_queue.append(qi)  # 依次加入元素
                #
                new_queue.append(date)  # 再添加到队尾
                new_queue.pop(0)  # 再去掉队头
                self.ways.append(new_queue)
                # 保存当前的部分和所出现的次数
                self.Sum_count[sum + int(date)] = self.Sum_count.get(sum + int(date),0) + 1
            elif sum + int(date) == self.TARGET: # 当等于当前的最大值时,存储之前不清空
                # 保存当前方案,注意:保证顺序不要乱
                new_queue = []
                new_queue.append(date)  # 先添加到队头
                for qi in queue:
                    new_queue.append(qi)  # 依次加入元素
                #
                new_queue.append(date)  # 再添加到队尾
                new_queue.pop(0)  # 再去掉队头
                self.ways.append(new_queue)
                # 保存当前的部分和所出现的次数
                self.Sum_count[sum + int(date)] = self.Sum_count.get(sum + int(date), 0) + 1
            ## 考虑一下,这里是否需要return
            ### 如果date长度等于1,则return;否则继续执行——
            if len(date) == 1:
                return
        #### 3
        # 将数据拆分成子串
        LEN = len(date)
        store = []
        ## 1 将数据划分为两个子串,对后一个子串进行DFS搜索
        for i in range(LEN - 1):
            store.append([date[:i + 1], date[i + 1:]])
        LEN = len(store)
        ## 对每一组子串进行分析处理
        for i in range(LEN):
            # 取出待处理的数据
            sub1 = store[i][0]
            sub2 = store[i][1]
            # 处理:
            sum = sum + int(sub1)  # 部分和sum
            # 判断部分和是否超过了目标
            if sum <= int(target):
                ##
                new_queue = []
                new_queue.append(sub1)  # 先添加到队头
                for qi in queue:
                    new_queue.append(qi)  # 依次加入元素
                new_queue.append(sub1)  # 再添加到队尾
                new_queue.pop(0)  # 再去掉队头
                ##
                self.dfs(sub2, sum, new_queue)
                # 再进行下一次搜索之前,恢复状态
                sum = sum - int(sub1)
                new_queue.pop()
            else:
                return

        return

## 这个题目,需要特别注意查询所有可行的方案
data = []
target_data = []
while True:
    n,m = map(str,input().strip().split())
    if n =='0' and m =='0':
        break
    else:
        target_data.append(n)
        data.append(m)
# print(data)
## Case总数
N = len(data)
for case in range(N):
    # 获得初始数据
    subData = data[case]
    target  = target_data[case]
    # 1 初始判断
    ##1.1 数据等于目标数据,直接输出
    if subData == target:
        print(target+' '+subData)
        continue
    ##1.2 逐个裁剪成单个数字
    len_str = len(subData)
    res_int = []
    res_str = ''
    sum = 0
    for str_i in range(len_str):
        res_int.append(int(subData[str_i]))
        res_str = res_str + ' ' + subData[str_i]
        sum = sum + int(subData[str_i])
    ###1.2.1 目标数据小于最小和,报错
    if int(target) < sum:
        print('error')
        continue
    ###1.2.2 目标数据等于最小和,且数据中无0元素
    if int(target) == sum and 0 not in res_int:
        print(str(sum)+' '+ res_str.strip())
    # 2 DFS处理数据
    test = Solutions()
    ways = test.DFS()
    if len(ways)>1:
        print('rejected')
    else:
        way = ways[0]
        ans_sum = 0
        res = ''
        for wi in way:
            res = res + ' ' + wi
            ans_sum = ans_sum + int(wi)
        print(str(ans_sum) + ' ' +res.strip())



版本一

# POJ 1416 - Shredding Company

# https://exp-blog.com/algorithm/poj/poj1416-shredding-company/
# https://blog.csdn.net/lyy289065406/article/details/6647969
import collections
class Solution:
    def DFS(self,target,subData):
        self.target = int(target)
        # 记录方案
        self.ways = collections.defaultdict(list)
        # 记录出现的Sum的类别及数目
        self.Sum_dict = collections.defaultdict(list)
        # 给定一个初始Sum值0,找到一个最大且不超过Target的Sum,并且记录该方案至self.ways
        # 初始Sum
        self.Sum_pre = 0
        self.dfs(subData,0,[],False)
        return self.ways

    def dfs(self,Data,Sum,parent,Flag):
        # 1 对于队列而言,首先将其转存为新的变量,便于实现回溯时的状态恢复
        new_parent = []
        for i in parent:
            new_parent.append(i)

        ## 2 判断是否终止
        if Flag == True:# 是否叠加到了最后一串字符呢?
            # 求和
            Sum = Sum + int(Data)
            if Sum >= self.Sum_pre and Sum<= self.target:
                # 更新获取更大的和
                self.Sum_pre = Sum
                # 保存当前的节点Sum
                self.Sum_dict[Sum] = self.Sum_dict.get(Sum,0) + 1; # 统计出现同一Sum的次数
                ## 将方案保存
                # new_parent.append(Data)
                temp_way = [] #为了防止new_parent引起self.ways更新,此处重新赋值
                for i in new_parent:
                    temp_way.append(i)
                temp_way.append(Data)
                self.ways[Sum].append(temp_way)
                ## 特别处理——当Data仍然可拆分时,需要进一步拆分
                if len(Data)>=2: #当Data仍然可拆分时,需要进一步拆分
                    # 这里要恢复new_parent
                    # new_parent.pop() #弹出最后加入的点
                    Sum = Sum - int(Data)  # 拆分之前,恢复数据
                else:
                    return # 如果没必要再分,就结束吧!
            else:
                Sum = Sum - int(Data) #恢复数据

        ## 3 循环
        E = len(Data)
        store = []
        for i in range(E-1): # 拆成前后两个字符串
            store.append([Data[:(i+1)],Data[i+1:]])

        for i in range(len(store)): # 对每一组字符串进行DFS搜索处理
            ## 第一段不递归 
            sub1 = store[i][0]
            Sum_new = Sum + int(sub1) # 这里需要重新赋值,以便于回溯时恢复参数状态
            if Sum_new > self.target: # 如果部分和超过了目标,则返回
                return
            ## 第二段递归
            new_parent.append(sub1)
            sub2 = store[i][1]
            self.dfs(sub2, Sum_new, new_parent, True)
            ## 奇葩的处理:这里不返回,而是继续执行下一次的循环
            # 循环之前,弹出节点,恢复一下列表参数
            new_parent.pop()
        return


## 迎接输入:
data = [] #利用队列存储目标和数据
while True:
    str1,str2 = input().strip().split(' ')
    if int(str1)==0 and int(str2)==0:
        break
    else:
        data.append([str1,str2])
# print(data)
# 共计N组数据
N = len(data)
# print(N)
for i in range(N):
    # 获得目标和数据
    target,subData = data[i][0],data[i][1]
    ## 第一级判别:目标与数据是否直接相等——是:打印数据
    if target == subData:
        print(target+' '+subData)
        continue # 去执行下一次循环
    ## 第二级判别:按元素个数划分,其和是否大于目标——
    # 数据元素个数
    E = len(subData)
    # print(E)
    # 计算E个元素之和sum
    E_sum = 0
    for i in range(E):
        E_sum = E_sum + int(subData[i])
    if E_sum > int(target):# 按元素个数划分,其和是否大于目标——是:打印error
        print("error")
        continue
    if E_sum == int(target) and '0' not in subData:# 按元素个数划分,其和是否等于目标,并且数据中无0元素——是:打印出来
        res = target
        for i in range(E):
            res = res + ' ' + subData[i]
        print(res)
        continue
    ## 接下里,利用DFS处理一般情形——
    test = Solution()
    ways = test.DFS(target,subData)
    # 处理输出结果
    MAX_Sum = max(ways.keys()); #查找接近于目标的最大和
    ans = ways[MAX_Sum] #获取相应的方案
    if len(ans)>1: # 当存在多种方案时
        print("rejected")
    else:
        # 按照格式打印输出
        ans_sum = 0
        res = ''
        for i in ans[0]:
            ans_sum += int(i)
            res = res + ' ' + i
        print(str(ans_sum) + ' ' + res.strip(' '))








输入:

50 12346
376 144139
927438 927438
18 3312
9 3142
25 1299
111 33333
103 862150
6 1104
0 0

输出:

43 1 2 34 6
283 144 139
927438 927438
18 3 3 12
error
21 1 2 9 9
rejected
103 86 2 15 0
rejected

这是一道比较经典的计数问题。题目描述如下: 给定一个 $n \times n$ 的网格图,其中一些格子被标记为障碍。一个连通块是指一些被标记为障碍的格子的集合,满足这些格子在网格图中连通。一个格子是连通的当且仅当它与另一个被标记为障碍的格子在网格图中有公共边。 现在,你需要计算在这个网格图中,有多少个不同的连通块,满足这个连通块的大小(即包含的格子数)恰好为 $k$。 这是一道比较经典的计数问题,一般可以通过计算生成函数的方法来解决。具体来说,我们可以定义一个生成函数 $F(x)$,其中 $[x^k]F(x)$ 表示大小为 $k$ 的连通块的个数。那么,我们可以考虑如何计算这个生成函数。 对于一个大小为 $k$ 的连通块,我们可以考虑它的形状。具体来说,我们可以考虑以该连通块的最左边、最上边的格子为起点,从上到下、从左到右遍历该连通块,把每个格子在该连通块中的相对位置记录下来。由于该连通块的大小为 $k$,因此这些相对位置一定是 $(x,y) \in [0,n-1]^2$ 中的 $k$ 个不同点。 现在,我们需要考虑如何计算这些点对应的连通块是否合法。具体来说,我们可以考虑从左到右、从上到下依次处理这些点,对于每个点 $(x,y)$,我们需要考虑它是否能够与左边的点和上边的点连通。具体来说,如果 $(x-1,y)$ 和 $(x,y)$ 都在该连通块中且它们在网格图中有公共边,那么它们就是连通的;同样,如果 $(x,y-1)$ 和 $(x,y)$ 都在该连通块中且它们在网格图中有公共边,那么它们也是连通的。如果 $(x,y)$ 与左边和上边的点都不连通,那么说明这个点不属于该连通块。 考虑到每个点最多只有两个方向需要检查,因此时间复杂度为 $O(n^2 k)$。不过,我们可以使用类似于矩阵乘法的思想,将这个过程优化到 $O(k^3)$ 的时间复杂度。 具体来说,我们可以设 $f_{i,j,k}$ 表示状态 $(i,j)$ 所代表的点在连通块中,且连通块的大小为 $k$ 的方案数。显然,对于一个合法的 $(i,j,k)$,我们可以考虑 $(i-1,j,k-1)$ 和 $(i,j-1,k-1)$ 这两个状态,然后把点 $(i,j)$ 加入到它们所代表的连通块中。因此,我们可以设计一个 $O(k^3)$ 的 DP 状态转移,计算 $f_{i,j,k}$。 具体来说,我们可以考虑枚举连通块所包含的最右边和最下边的格子的坐标 $(x,y)$,然后计算 $f_{x,y,k}$。对于一个合法的 $(x,y,k)$,我们可以考虑将 $(x,y)$ 所代表的点加入到 $(x-1,y,k-1)$ 和 $(x,y-1,k-1)$ 所代表的连通块中。不过,这里需要注意一个细节:如果 $(x-1,y)$ 和 $(x,y)$ 在网格图中没有相邻边,那么它们不能算作连通的。因此,我们需要特判这个情况。 最终,$f_{n,n,k}$ 就是大小为 $k$ 的连通块的个数,时间复杂度为 $O(n^2 k + k^3)$。 参考代码:
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值