T1212 仙岛求药——dfs 剪枝

T1212 仙岛求药——dfs 剪枝_第1张图片
T1212 仙岛求药——dfs 剪枝_第2张图片

分析

此题需要剪枝才能过,不然超时,多用了一个dis数组去记录到达每个点最少经过次数,如果当前cnt>dis,或者cnt已经比已得到临时答案大了,可以直接return;这种搜索的思路比较简单,和红与黑差不多;最短路问题建议使用bfs,因为dfs重复的情况太多,容易超时。

#include

using namespace std;

int m, n;
int dx[] = {-1, 0, 1, 0};
int dy[] = {0, 1, 0, -1};
char c[25][25];
int vis[25][25];
int dis[25][25];//到某个点最少经过点数
int ans = 100;

void dfs(int x, int y, int cnt) {
    if (cnt >= ans || cnt > dis[x][y])//剪枝
        return;
    if (c[x][y] == '*') {
        //路径可能多条,选择最优
        ans = min(ans, cnt);
        return;
    }
    dis[x][y] = min(cnt, dis[x][y]);
    for (int i = 0; i < 4; i++) {
        int xx = x + dx[i];
        int yy = y + dy[i];
        if (!vis[xx][yy] && !(c[xx][yy] == '#' || c[xx][yy] == '@') && xx >= 1 && yy >= 1 && xx <= m && yy <= n) {
            vis[xx][yy] = 1;
            dfs(xx, yy, cnt + 1);
            vis[xx][yy] = 0;
        }
    }
}

int main() {
    cin.tie(nullptr);
    cin >> m >> n;
    memset(dis, 0x3f3f3f3f, sizeof dis);
    int x, y;
    for (int i = 1; i <= m; i++) {
        for (int j = 1; j <= n; ++j) {
            cin >> c[i][j];
            if (c[i][j] == '@') {
                x = i, y = j;
            }
        }
    }
    vis[x][y] = 1;
    dfs(x, y, 0);
    if (ans != 100)
        cout << ans << endl;
    else
        cout << -1 << endl;
    return 0;
}

你可能感兴趣的:(dfs,计蒜客,深度优先,剪枝,算法)