「AtCoder Beginner Contest 253B - Distance Between Tokens」メモ

問題

AtCoder Beginner Contest 253

https://atcoder.jp/contests/abc253/tasks/abc253_b

考察

oの位置を記録する配列をつくる。

ans.append(i,j)

で一つの部分に二個のデータをまとめた配列がつくれる。

ACコード

def Tokens(a,h,w):
    ans = []
    for i in range(h):
        for j in range(w):
            if a[i][j] == "o":
                ans.append((i,j))
    return ans

h,w=map(int,input().split())

s= [input() for _ in range(h)]

ans = Tokens(s,h,w)

a,c = ans[0]
b,d = ans[1]

print(abs(a-b)+abs(c-d))