Crazy Robot(dfs)

Crazy Robot

题意

Problem - 1613E - Codeforces从L开始找到一条路径只有一或者两个开口。

思路

dfs，对于每个合法节点看其周边的封堵的方向个数封堵的方向个数必须>=3才可以，从’L’开始搜。本题n*m<=1e6，不能开二维数组，用string[]或者vector，vis标记使用了set。

#include <bits/stdc++.h>
typedef long double ld;
typedef long long ll;
#define pb push_back
#define mk make_pair
#define mt make_tuple
#define eb emplace_back
#define pob pop_back
#define rz resize
#define mem(a,b) memset(a,b,sizeof(a))
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()
#define debug(a) cout<<#a<<"="<<a<<endl;
#define xx first
#define yy second
#define qwe(i,a,b) for(int i=a;i<=b;i++)
#define ewq(i,a,b) for(int i=a;i>=b;i--)
inline ll rr(){ll f=1,x=0;char ch;do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');return f*x;}
using namespace std;
// ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
const ll INF=0x3f3f3f3f,inf=0x3f3f3f3f3f3f3f;
const ll mod[2]={int(1e9 + 7), 10007};
const int base[2]={29,31};
const int maxn=1e6+6;
ll qpow(ll x,ll n){ll ans=1;while(n>0){if(n&1)ans=ans*x%mod[1];x=x*x%mod[1];n>>=1;}return ans;}

int n,m,xi,yi;
string s[maxn];
set<int> a[maxn];
int v[][2]={{-1,0},{1,0},{0,-1},{0,1}};
struct node {
    int x,y;
};
bool in_area(int x,int y) {
    return x>=0 && x<n && y>=0 && y<m;
}
void dfs(int x,int y) {
    for(int i=0;i<4;i++) {
        int tx=x+v[i][0],ty=y+v[i][1];
        if(!in_area(tx,ty) || s[tx][ty]!='.') continue;
        if(a[tx].count(ty)) continue;
        int cnt=0;
        for(int j=0;j<4;j++) {
            int cx=tx+v[j][0],cy=ty+v[j][1];
            if(!in_area(cx,cy) || s[cx][cy]=='#' || s[cx][cy]=='+' || s[cx][cy]=='L')
                cnt++;
        }
        if(cnt>=3) {
            s[tx][ty]='+';
            a[tx].insert(ty);
            dfs(tx,ty);
        }
    }
}
int main(int argc, char const *argv[]) {
    int t=rr();
    while (t--) {
        n=rr(),m=rr();
        qwe(i,0,n-1) a[i].clear();
        int fl=1;
        qwe(i,0,n-1) {
            cin>>s[i];
            if(fl) qwe(j,0,m-1) if(s[i][j]=='L') xi=i,yi=j,fl=0;
        }
        dfs(xi,yi);
        qwe(i,0,n-1) {
            std::cout << s[i] << '\n';
        }
    }
    return 0;
}