代码随想录：103. 水流问题

103. 水流问题

这道题采用遍历四边逆流，判断两组边界共同到达的点位，然后输出

1、条件准备

firstvisit存第一组边界出发能到达的点，endvisit存第二组边界出发能到达的点。

graph数组存图，n为行数，m为列数

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = l; i <= r; i++)
using namespace std;

#define endl '\n'

int dir[4][2]={0,1,1,0,0,-1,-1,0};
vector<vector<int>> firstvisit(105,vector<int>(105,0));
vector<vector<int>> endvisit(105,vector<int>(105,0));
int graph[105][105];
int n,m;

先存图，然后分别遍历四边，看看能到达哪里。

最后把都能到达的输出即可

int main()
{
  std::ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cin>>n>>m;
  rep(i,0,n-1)
    rep(j,0,m-1)
    cin>>graph[i][j];
 
rep(i,0,n-1)
  {
    dfs(i,0,firstvisit);
    dfs(i,m-1,endvisit);
  }

  rep(i,0,m-1)
  {
    dfs(0,i,firstvisit);
    dfs(n-1,i,endvisit);
  }

  rep(i,0,n-1)
  rep(j,0,m-1)
   if(firstvisit[i][j]&&endvisit[i][j])cout<<i<<' '<<j<<endl;

  return 0;
}

2、dfs函数

遍历四周，若小于等于周围的点，则遍历它。走过的点置1.

void dfs(int x,int y,vector<vector<int>> &visit)
{
        if(visit[x][y])return ;
        visit[x][y]=1;

        for(int i=0;i<4;i++)
        {
            int xx=x+dir[i][0],yy=y+dir[i][1];
            if(xx<0||yy<0||xx>=n||yy>=m)continue;
            if(graph[xx][yy]>=graph[x][y])
              dfs(xx,yy,visit);
        }
}

3、总结

这个题主要是可以从四周遍历来dfs也是常用的一种方法

完整代码如下：

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = l; i <= r; i++)
using namespace std;

#define endl '\n'

int dir[4][2]={0,1,1,0,0,-1,-1,0};
vector<vector<int>> firstvisit(105,vector<int>(105,0));
vector<vector<int>> endvisit(105,vector<int>(105,0));
int graph[105][105];
int n,m;

void dfs(int x,int y,vector<vector<int>> &visit)
{
        if(visit[x][y])return ;
        visit[x][y]=1;

        for(int i=0;i<4;i++)
        {
            int xx=x+dir[i][0],yy=y+dir[i][1];
            if(xx<0||yy<0||xx>=n||yy>=m)continue;
            if(graph[xx][yy]>=graph[x][y])
              dfs(xx,yy,visit);
        }
}

int main()
{
  std::ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cin>>n>>m;
  rep(i,0,n-1)
    rep(j,0,m-1)
    cin>>graph[i][j];
 
rep(i,0,n-1)
  {
    dfs(i,0,firstvisit);
    dfs(i,m-1,endvisit);
  }

  rep(i,0,m-1)
  {
    dfs(0,i,firstvisit);
    dfs(n-1,i,endvisit);
  }

  rep(i,0,n-1)
  rep(j,0,m-1)
   if(firstvisit[i][j]&&endvisit[i][j])cout<<i<<' '<<j<<endl;

  return 0;
}