pandasでpaiza問題(陣取りゲーム:1 マスの陣取り)

スポンサーリンク

paizaへのリンク(陣取りゲーム:1 マスの陣取り (paizaランク C 相当)

flagを立てているが、無くても同じ動きをする。ただ、速度が少し違うはず。

import pandas as pd

def main():

    h, w = map(int, input().split())
    lst1 = []

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

    df1 = pd.DataFrame(lst1)
    flag = 0

    for i in range(h):
        if flag == 1:
            break
        for j in range(w):
            if flag == 1:
                break
            if df1.iat[i, j] == '*':
                if i > 0:
                    df1.iat[i - 1, j] = '*'
                if i < h - 1:
                    df1.iat[i + 1, j] = '*'
                if j > 0:
                    df1.iat[i, j - 1] = '*'
                if j < w - 1:
                    df1.iat[i, j + 1] = '*'
                flag = 1

    for i in range(h):
        for j in range(w):
            print(df1.iat[i, j], end='')
        print()

if __name__ == '__main__':
    main()

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

コメント

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

Please disable your adblocker or whitelist this site!

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