算法题(镜子田地)

145 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第18天,点击查看活动详情

镜子田地

题目描述

农夫约翰在屋子外面放了一些旧镜子,他的奶牛们像往常一样调皮地偷走了它们!

奶牛们将镜子放置在了一个矩形田地中,该田地可被划分为 N×MN×M 个方格区域。

在每个方格区域中,奶牛在其某对对角之间放置一个双面镜,因此,共有两种放法,一种为 / 放置(镜子连接方格左下角和右上角),另一种为 `` 放置(镜子连接方格左上角和右下角)。

一天晚上,奶牛贝茜将激光发射器带到了该田地中。

她站在田地外面,沿着田地的行或列水平或垂直照射光束,使光束反射一定数量的镜子。

由于镜子都是沿对角线摆放,因此经反射镜反射的水平光束最终将垂直传播,反之亦然。

贝茜想知道从田地之外射入的水平或垂直光束最多可以在田地中被反射多少次。

给定镜子田地的布局,请帮助贝茜计算这个数字。

输入格式

第一行包含 NN 和 MM。

接下来 NN 行,每行包含 MM 个 / 或 `` 字符,表示田地中镜子的具体摆放方式。

输出格式

输出田地之外的水平或垂直光束能够被反射的最大次数。

如果可以无限反射,则输出 −1−1。

数据范围

1≤N,M≤10001≤N,M≤1000

输入样例:

3 3
/\
\\
//

输出样例:

3

样例解释

贝茜可以从上向下沿中间列上方发射激光。

共可以反射 33 次。

代码

#include<bits/stdc++.h>
using namespace std;

int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};//分别是:向上,向右,向下,向左
char g[1010][1010];
int ans, n, m;

void dfs(int x, int y, int d, int u){
    ans = max(ans, u);
    int nd;
    if(g[x][y] == '/') nd = d ^ 1;
    else nd = d ^ 3;

    int nx = x + dx[nd], ny = y + dy[nd];//下一个位置

    if(nx && ny && nx <= n && ny <= m)//如果下一个位置依旧在规定位置里
        dfs(nx, ny, nd, u + 1);
}

int main(){
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            cin >> g[i][j];

    for(int i = 1; i <= m; i++){
        dfs(1, i, 2, 1);//最上面的一行,入射方向向下
        dfs(n, i, 0, 1);//最下面的一行,入射方向向上
    }
    for(int i = 1; i <= n; i++){
        dfs(i, 1, 1, 1);//最左边的一行,入射方向向右
        dfs(i, m, 3, 1);//最右边的一行,入射方向向左
    }

    cout << ans;
    return 0;
}