pandasでpaiza問題(マップの判定・縦横:マップの判定・横)

スポンサーリンク

paizaへのリンク(マップの判定・縦横:マップの判定・横 (paizaランク C 相当)

ここではリストの端にいるか否かをif文で判定しているが、リスト全体を’.’で一回り囲うという別解もありえる。

import pandas as pd


def main():
    h, w = [int(_) for _ in input().split()]
    lst1 = []

    for i in range(h):
        lst1.append(list(input()))

    df1 = pd.DataFrame(lst1)

    for i in range(h):
        for j in range(w):
            counter = 0

            if j == 0:
                counter += 1
            elif df1.iat[i, j - 1] == '#':
                counter += 1

            if j == w - 1:
                counter += 1
            elif df1.iat[i, j + 1] == '#':
                counter += 1

            if counter == 2:
                print(str(i) + ' ' + str(j))


if __name__ == '__main__':
    main()

「pandasでpaiza問題」記事一覧へのリンク

コメント

当ブログのコンテンツが気に入ったら広告ブロックの解除(ホワイトリスト化)をご検討下さい。

Please disable your adblocker or whitelist this site!

タイトルとURLをコピーしました