[프로그래머스 코딩테스트 입문] - [공 던지기, 배열 회전시키기]

공 던지기 

문제 설명
머쓱이는 친구들과 동그랗게 서서 공 던지기 게임을 하고 있습니다. 

공은 1번부터 던지며 오른쪽으로 한 명을 건너뛰고 그다음 사람에게만 던질 수 있습니다. 

친구들의 번호가 들어있는 정수 배열 numbers와 정수 K가 주어질 때, 

k번째로 공을 던지는 사람의 번호는 무엇인지 return 하도록 solution 함수를 완성해보세요.
제한사항
2 < numbers의 길이 < 100
0 < k < 1,000
numbers의 첫 번째와 마지막 번호는 실제로 바로 옆에 있습니다.
numbers는 1부터 시작하며 번호는 순서대로 올라갑니다.
입출력 예
numbers               k      result
[1, 2, 3, 4]             2         3
[1, 2, 3, 4, 5, 6]     5         3

코드

풀이1
def solution(numbers, k):
    count = 0       # numbers 배열의 인덱스값
    num = 1 		# 던지는 횟수
    while(True):
        if k == num:
            answer = numbers[count]
            break
        # 공을 던지는 순서가 배열의 마지막, 마지막-1값이 아닐때    
        if count < len(numbers) - 2:  
            count += 2
            num += 1  
        # 공을 던지는 순서가 배열의 마지막-1일때    
        elif ((len(numbers)) - 2)  == count:  
            num += 1
            count = 0
        # 공을 던지는 순서가 배열의 마지막일때    
        else:
            num += 1
            count = 1
    return answer
    
    
 풀이2
 def solution(numbers, k):
    return numbers[2 * (k - 1) % len(numbers)]

 

 

배열 회전시키기

문제 설명
정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다. 

배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을 return
제한사항
3 ≤ numbers의 길이 ≤ 20
direction은 "left" 와 "right" 둘 중 하나입니다.
입출력 예
numbers      direction    result
[1, 2, 3]         "right"      [3, 1, 2]

풀이

def solution(numbers, direction):
    answer = []
    a = ''
    if direction == 'right':
        a = numbers[int(len(numbers)-1)]
        numbers.pop()
        numbers.insert(0,a)
        answer = numbers
    else:
        a = numbers[0]
        del numbers[0]
        numbers.append(a)
        answer = numbers
    return answer

a = [1,2,3] 

추가

- a.insert(0, 1) : 리스트의 맨 처음에 요소(1) 추가

- a.insert(len(a), 5) = a.append(5) : 리스트 끝에 요소(5) 추가

- a[1:1] = [4,5] -> a = [1,4,5,2,3] : 인덱스 1부터 4,5 추가 (슬라이스사용)

삭제

- pop() 리스트 마지막요소 삭제

- 인덱스로 요소 삭제 : a.pop(2) = del a[2]

- 특정 값 삭제 : a.remove(20)