파이썬 1,2, 다차원 배열 입력 받기

1 2 3 4 5 6 7 8 9 10 11 12 13 14

# 이렇게 아래처럼 사용하면 2차원 배열 생성 가능 N = int(input()) #width of square grid grid = [list(map(int, input())) for _ in range(N)] print(grid)   # map을 이용하여 빠르게 배열 리스트 생성 가능 arr1 = map(int, input().split()) arr2 = map(int, input().split()) print(N) for i in arr1:     print(i) for i in arr2:     print(i)   Colored by Color Scripter

cs

gird 부분에서는 n번동안 map(int, input())를 list에 받아내기에 2차원이 가능하다.

arr1, arr2 = map(int, input().split())로 하면 int n, m에 각각 scanf 받는것과 동일

n = int(input())
# 배열을 아래와 같이 받으면 빠르다.
tmp = map(int, input().split())
 
'''
문제를 해결하기 위해 그리디하게 생각하자
각 숫자를 해당하는 arr 2차원 리스트 인덱스에 tmp에 해당하는 인덱스 번호를 넣어준다.

1 1 2 2 3 3이면 arr 리스트 1번 인덱스에 1,2를 append 한다는 것이다
그러면 먼저 나온 번호부터 차례대로 정렬이 되게 되는데

arr의 1번 인덱스 첫번째 것과 2번 인덱스 첫번째 것의 절댓값 차를 계속 더해가면서
정답을 구해주면 된다.

설명이 어렵지만 코드를 보면 쉽게 해결 할 수 있다.
'''
arr = [[] for i in range(n + 1)]
arr[0].append(1)
arr[0].append(1)
 
idx = 1
for i in tmp:
    arr[i].append(idx)
    idx += 1
 
sum1 = 0
sum2 = 0
for i in range(1,n + 1):
    sum1 += abs(arr[i][0] - arr[i - 1][0])
    sum2 += abs(arr[i][1] - arr[i - 1][1])
print(sum1 + sum2)