这一题的dfs剪枝怎么做

P1216 [USACO1.5] [IOI1994]数字三角形 Number Triangles

乌丑 @ 2021-12-12 20:12:14

老师让我们用dfs,又把最大路径和改成最小路径和,于是我给出了如下代码

#include<bits/stdc++.h>
using namespace std;
int a[100][100],n;
int dfs(int x,int y) {
    if(x==n) return a[x][y];
    int left=dfs(x+1,y),right=dfs(x+1,y+1);
    return min(left,right)+a[x][y];
}
int main() {
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=i;j++) cin>>a[i][j];
    cout<<dfs(1,1);
    return 0;
}

但是老师考虑到会有大量多次计算,让我剪枝,然后就不会了, 比如:
红色圈中的数在left中被算了一次,在right中又被算了一次,如何避免??


by Y2y7m @ 2021-12-12 20:16:10

非要用搜索吗,动规他不香吗?


by 45dino @ 2021-12-14 21:55:03

@乌丑 可以考虑记忆化

对于每个点存储一个值,表示前面到这个点经过的数的总和的最小值。

假如某一次dfs又到达了这个点,就先拿这个值与本次dfs的和进行比较,假如存下来的这个值更小,就没有必要继续下去。


by newbie_QwQ @ 2021-12-14 21:55:56

使用记忆化搜索。

#include<iostream>
using namespace std;
int bs[1001][1001],zd[1001][1001];
int dfs(int x,int y,int n)
{
    if(x==n) return bs[x][y];
    if(zd[x][y]!=-1) return zd[x][y];
    return zd[x][y]=max(dfs(x+1,y,n),dfs(x+1,y+1,n))+bs[x][y]; 
}
int main()
{
    int i,j,n;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=i;j++)
        {
            cin>>bs[i][j];
            zd[i][j]=-1;
        }
    }
    cout<<dfs(1,1,n);
    return 0;
}

by newbie_QwQ @ 2021-12-14 21:56:38

拿空间换时间就可以了。


by tobie @ 2021-12-14 21:59:36

@乌丑 您可以设置一个数组f[x][y]初始值设为-1

搜索时如果f[x][y]==-1 则没有被遍历过按照正常流程走,同时用算出的答案去更新f[x][y](告诉程序dfs(x,y)已经算过了,下一次遇到后可以直接返回f[x][y]中存的数);

如果f[x][y]!=-1,则说明dfs(x,y)已经被算过了,直接返回f[x][y]即可。

int a[100][100],n;
int dp[100][100];
int dfs(int x,int y) {
    if(dp[x][y]!=-1) return dp[x][y];
    if(x==n) return a[x][y];
    int left=dfs(x+1,y),right=dfs(x+1,y+1);
    return dp[x][y]=min(left,right)+a[x][y];
}

|