链接:https://ac.nowcoder.com/acm/contest/331/I
来源:牛客网
题目描述
小希拿到了一个镜子块,镜子块可以视为一个N x M的方格图,里面每个格子仅可能安装`\`或者`/`的镜子,会反射90°光线,也可能没有安装镜子,使用`.`代替。
但她看不清楚里面的镜子构造是怎样的。
你是这块镜子块的主人,所以你想计算这块镜子块(从输入的上方往下射入光线)从左到右每一格射入依次分别会从最下面的哪一格子射出,如果无法射出,输出-1。
但她看不清楚里面的镜子构造是怎样的。
你是这块镜子块的主人,所以你想计算这块镜子块(从输入的上方往下射入光线)从左到右每一格射入依次分别会从最下面的哪一格子射出,如果无法射出,输出-1。
输入描述:
第一行输入两个整数N,M。随后的N行,每行M个字符。
1≤N,M≤5001≤N,M≤500
输出描述:
输出M行整数,依次为从第i个格子从上往下射入时从下面的哪里射出,如果光线不会从下面射出,则输出-1。
示例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; }