TIL 코딩공장

12/ 14 파이썬 공부
백준 1551번 수열의 변화

1) 문제
크기가 N인 수열 A가 주어졌을 때, 세준이는 인접한 두 원소의 차이를 이용해서 크기가 N-1인 수열 B를 만들 수 있다.

예를 들어, A = {5, 6, 3, 9, -1} 이었을 때, B = {6-5, 3-6, 9-3, -1-9} = {1, -3, 6, -10}이 된다. 
즉, B[i] = A[i+1]-A[i]가 된다.

수열 A가 주어졌을 때, 세준이가 위의 방법을 K번 했을 때 나오는 수열을 구하는 프로그램을 작성하시오.

2) 입력
첫째 줄에 수열의 크기 N과 K가 주어진다. 
N은 20보다 작거나 같은 자연수이고, K는 0보다 크거나 같고, N-1보다 작거나 같은 정수이다. 
둘째 줄에는 수열이 ‘,’로 구분되어 주어진다. 
수열을 이루고 있는 수는 절댓값이 100보다 작거나 같은 정수이다.

3) 출력
첫째 줄에 K번 변형한 수열을 ‘,’로 구분하여 출력한다.

>>>코드1. index, Type 오류

n, k = map(int, input().split())
l1 = [map(int, input().split(','))]
for i in range(k):
    l2 = []
    for j in range(n-i-1):
        l2.append(l1[j+1]-l1[j])
    l1 = l2.copy()
print(','.join(l1))

- 리스트를 입력받을 때, [] 대신 list()를 이용해서 입력받아야 index error가 나지 않는다.

[]의 경우 코드의 속도는 더 빠르지만, 중간의 단계가 한단계 적다고 한다.

(1) []
- 0 BUILD_LIST : [ ]를 바로 인지하고 LIST함수를 적용합니다.
- 2 RETURN_VALUE : 말 그대로 값을 Return합니다.

(2) list
- 0 LOAD_NAME : list라는 이름을 존재하는 수많은 변수 중에서 찾습니다. 
그리고 찾는 순서는 Local Scope -> Enclosing Scope -> Global Scope -> Built-in Scope 라고 합니다. 
- 2 CALL_FUNCTION : 이제 list()라는 함수와 매칭이 되는것을 찾았을테니 함수를 수행합니다.
만약에 매치하는 함수가 어디에도 없다면 그런이름을 가진 변수는 없다는 에러가 출력되겠지요!
- 4 RETURN_VALUE : 말 그대로 값을 Return합니다.

출처: https://conansjh20.tistory.com/79 [취미는 파이썬:티스토리]

근데 이것보단, 파이썬에서 map() 함수는 맵 객체를 반환한다. 
이는 iterable한 객체이지만, 명시적으로 리스트나 다른 자료형으로 변환되지 않은 경우 리스트로 취급되지 않는다.

따라서, l1 = list(map(int, input().split(',')))와 같이 list()를 사용하여 
map()의 결과를 명시적으로 리스트로 변환하는 것이 중요하다. 
이렇게 해야 l1은 실제 리스트가 되며, 리스트로 예상되는 반복문 처리가 올바르게 수행된다.

그렇기 때문에 [map(int, input().split(','))] 대신에 
list(map(int, input().split(',')))와 같이 list()를 사용하여 리스트로 변환해야 올바르게 작동한다.

- .join()함수는 리스트 속 요소들이 문자열일 경우에만 사용 가능

>>>코드2. 리스트 입력과 join함수를 고친 코드

n, k = map(int, input().split())
l1 = list(map(int, input().split(',')))
for i in range(k):
    l2 = []
    for j in range(n-i-1):
        l2.append(l1[j+1]-l1[j])
    l1 = l2.copy()
for i in range(len(l1)):
    print(l1[i], end = '')
    if i != len(l1)-1:
        print(',', end = '')

>>>문제 링크
https://www.acmicpc.net/problem/1551