[알고리즘문제풀기] 공 던지기

문제

내가 작성한 정답

class Solution {
    public int solution(int[] numbers, int k) {
        return numbers[(2*(k-1))%numbers.length];
    }
}

다른 사람들의 정답

class Solution {
    public int solution(int[] numbers, int k) {
        return (k-1)*2 % numbers.length+1;
    }
}

Queue 이용

import java.util.*;

class Solution {
    public int solution(int[] numbers, int k) {
        int answer = 0;

        Queue<Integer> queue = new LinkedList<>();
        for(int num : numbers) queue.add(num);

        int cnt = 1;
        
        // queue.poll()은 큐의 맨 앞 요소를 제거하고 반환.
        // queue.add(queue.poll())는 이 요소를 큐의 맨 뒤로 이동시킴.
        // 두 번의 poll()을 통해 두 개의 요소를 큐의 뒤로 이동시키고, cnt를 증가시킨다.
        while(cnt!=k){
            queue.add(queue.poll());
            queue.add(queue.poll());
            cnt++;
        }
        answer = queue.poll();
        return answer;
    }
}