1. 问题描述:
给定一个 n × n 的字符矩阵,表示一片海域。矩阵中 # 表示暗礁区域,. 表示安全区域。现在要将一个 1 × k 的战舰投放到海域中。投放时,战舰不可接触到暗礁区域。战舰可以横着投放,也可以竖着投放。在投放完成后,每个安全区域要么包含战舰的一部分,要么不包含。对于某个安全区域,如果所有可能的投放方式中,共有 m 种不同的投放方式,满足该区域包含战舰的一部分,那么就称该区域的投放指数为 m。请确定投放指数最大的安全区域的位置坐标。
输入格式
第一行包含两个整数 n 和 k。接下来 n 行,每行包含一个长度为 n 的字符串,表示海域。
输出格式
输出投放指数最大的安全区域的行、列坐标。行从上到下,从 1 开始计数,列从左到右,从 1 开始计数。如果答案不唯一,输出任意合理方案均可。如果海域中根本无法投放战舰,则输出任意区域均可。
数据范围
1 ≤ k ≤ n ≤ 100
输入样例1:
4 3
#..#
#.#.
....
.###
输出样例1:
3 2
输入样例2:
10 4
#....##...
.#...#....
..#..#..#.
...#.#....
.#..##.#..
.....#...#
...#.##...
.#...#.#..
.....#..#.
...#.#...#
输出样例2:
6 1
输入样例3:
19 6
##..............###
#......#####.....##
.....#########.....
....###########....
...#############...
..###############..
.#################.
.#################.
.#################.
.#################.
#####....##....####
####............###
####............###
#####...####...####
.#####..####..#####
...###........###..
....###########....
.........##........
#.................#
输出样例3:
1 8
来源:https://www.acwing.com/problem/content/description/3788/
2. 思路分析:
分析题目可以知道我们需要找到这样一个位置,这个位置能够尽可能覆盖最多长度为k的区间,如果存在那么输出这个位置,否则输出任意一个位置即可。因为数据规模不是特别大所以可以枚举所有可能的位置,如果当前位置是"."那么我们需要找到包含当前这个位置左右两边最远的位置求解出对应的方案数目,同理求解出包含当前这个位置上下两边最远的位置求解出对应的方案数目两者相加就是包含这个位置所有长度为k的区间的方案,然后更新一下答案和对应的坐标即可。
3. 代码如下:
class Solution:
def process(self):
n, k = map(int, input().split())
# 多声明一个长度这样g中元素可以从下标为1的位置开始, 下面是第一行的元素
g = [["."] * (n + 1)]
for i in range(n):
# 前面添加一个元素"."这样元素下标可以从1的位置开始方便处理
g.append(["."] + list(input()))
res = 0
# x, y为当前投放指数最大的的安全区域的位置坐标
x = y = 1
# 枚举所有可能的结果
for i in range(1, n + 1):
for j in range(1, n + 1):
if g[i][j] == "#": continue
# 枚举当前位置左边可以到达的最远的位置
l = r = j
t = 0
# 枚举找到当前位置左右两边能够走的最远的位置, 因为要包含当前的位置所以需要限制r - j < k的判断
while r <= n and r - j < k and g[i][r] == ".": r += 1
while l >= 1 and j - l < k and g[i][l] == ".": l -= 1
# l, r是安全区域的上下一个位置所以为r - l - k(其实是相互抵消了两个位置)
t += max(0, r - l - k)
l = r = i
# 枚举找到当前位置上下可以到达的最远的位置(与左右两边是一样的)
while l >= 1 and i - l < k and g[l][j] == ".": l -= 1
while r <= n and r - i < k and g[r][j] == ".": r += 1
t += max(0, r - l - k)
if t > res:
# 更新坐标和答案
res = t
x, y = i, j
print(x, y)
if __name__ == '__main__':
Solution().process()