牛客寒假算法基础集训营5 I 炫酷镜子

链接：https://ac.nowcoder.com/acm/contest/331/I
来源：牛客网

题目描述

小希拿到了一个镜子块，镜子块可以视为一个N x M的方格图，里面每个格子仅可能安装`\`或者`/`的镜子,会反射90°光线，也可能没有安装镜子，使用`.`代替。

但她看不清楚里面的镜子构造是怎样的。

你是这块镜子块的主人，所以你想计算这块镜子块（从输入的上方往下射入光线）从左到右每一格射入依次分别会从最下面的哪一格子射出，如果无法射出，输出-1。

输入描述:

第一行输入两个整数N,M。随后的N行，每行M个字符。

$\mathrm{1\le N,M\le 5001\le N,M\le 500}$

输出描述:

输出M行整数，依次为从第i个格子从上往下射入时从下面的哪里射出，如果光线不会从下面射出，则输出-1。

示例1

输入

复制

3 3
...
...
\.\

输出

复制

3
2
-1

说明

第一列遇到镜子两次反弹通过第三列射出。

第二列直接射出。

第三列因为镜子反弹后向右边射出。
感觉用个搜索就行了
注意在遇到镜子的时候换移动方向就行
AC代码

#include<bits/stdc++.h>
using namespace std;
char ma[501][501];
int n,m;
int f[4][2]={1,0/*下*/,0,1/*右*/,0,-1/*左*/,-1,0/*上*/};
int dfs(int x,int y)
{
    int k=0;
    while(x<n&&y<m&&x>=0&&y>=0)
    {
        if(ma[x][y]=='\\')
        {
            if(k==0) k=1;
            else if(k==1) k=0;
            else if(k==2) k=3;
            else k=2;
        }
        if(ma[x][y]=='/')
        {
            if(k==0) k=2;
            else if(k==1) k=3;
            else if (k==2) k=0;
            else k=1;
        }
        x+=f[k][0];
        y+=f[k][1];
    }
    if(x==n)
        return y+1;
    else return -1;
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
        cin>>ma[i][j];
    for(int i=0;i<m;i++)
        cout<<dfs(0,i)<<endl;
    return 0;
}