[Cos Pro 1급] 6차 4번 - 카드 뭉치

문제

 

1부터 n까지 숫자가 적힌 카드 뭉치를 다음 규칙에 따라 카드 뭉치를 섞었을 때, 아래에서부터 k번째에 있는 카드에 적힌 숫자를 알아내려합니다. 카드 뭉치 가장 아래에는 1이 쓰여있으며, 위로 갈수록 숫자가 커집니다. 카드 뭉치를 섞는 규칙은 다음과 같습니다.

 

1. 카드 뭉치를 정확히 반으로 나눕니다.

2. 두 개로 나뉜 카드 뭉치를 교차로 섞어서 카드 뭉치 하나를 만듭니다. 이때, 항상 1이 적힌 카드가 있는 뭉치를 아래로 놓고 섞습니다.

3. 위 과정을 반복합니다.

 

예를 들어, 1부터 6까지 숫자가 적힌 카드 뭉치를 3번 섞는다면 과정별로 아래에서부터 적힌 숫자는 다음과 같습니다.

 

| 횟수 | 카드 뭉치          |

|------|--------------------|

| 1    | [1, 4, 2, 5, 3, 6] |

| 2    | [1, 5, 4, 3, 2, 6] |

| 3    | [1, 3, 5, 2, 4, 6] |

 

이때, 아래에서 3번째에 있는 카드에 적힌 숫자는 5입니다.

처음에 가장 위에 있는 카드에 적힌 숫자 n, 섞는 횟수 mix, 알고 싶은 카드의 위치 k가 매개변수로 주어질 때, 아래에서부터 k번째에 있는 카드에 적힌 숫자를 return 하도록 solution 함수를 작성하세요.

 

 

 

매개변수 설명

처음에 가장 위에 있는 카드에 적힌 숫자 n, 섞는 횟수 mix, 알고 싶은 카드의 위치 k가 solution 함수의 매개변수로 주어집니다.

* n은 10 이상 50 이하인 짝수입니다.

* 카드는 0회 이상 10회 이하로 섞습니다.

* k는 1 이상 n 이하인 자연수입니다.

 

 

return 값 설명

아래에서 k번째에 있는 카드에 적힌 숫자를 return 해주세요.

 

예시

 

n	mix	k	return
6	3	3	5

 

Python

def solution(n, mix, k):
    cards = [ i for i in range(1,n+1) ]
    half = n//2
    for i in range(mix):
        left = cards[:half]
        right = cards[half:]
        
        for i in range(half):
            cards[2*i] = left[i]
            cards[2*i+1] = right[i]
    
    return cards[k-1]

print(solution(6,3,3))