BOJ - 달팽이 리스트 (17827)

https://www.acmicpc.net/problem/17827

17827번: 달팽이 리스트

 

 

#include <bits/stdc++.h>
 
using namespace std;
 
int main() {
    int n, m, v;
    cin >> n >> m >> v;
 
    vector<int> vv(n);
    for (int i = 0; i < n; i++) {
        cin >> vv[i];
    }
 
    int before_cycle = v - 2;    // 사이클 전까지 배열에서의 인덱스를 구함, 
                                // v가 3일 때 3번째 노드(인덱스 2 노드)도 사이클의 일부이다. 
                                // 따라서 1번 인덱스 노드까지가 사이클에 들어가기 전이므로 v - 2 를 해줌
 
    int cycle_length = n - (v - 1);    // 사이클 전까지 노드 갯수가 v - 1 개 이므로 n 에서 이를 빼준 값이 사이클 길이다.
 
    while (m--) {
        int q;
        cin >> q;
        // 경우의 수는 총 2가지
        // 1. 사이클 들어가기 전 인덱스 노드의 값을 구할 경우
        // 2. 사이클 들어간 인덱스 노드의 값을 구할 경우
        if (q <= before_cycle) {
            cout << vv[q] << endl;
        }
        else {
            // 사이클에 들어가고 나서 몇 번을 가는지 정확히 계산하기 위해서 구해줌
            q = q - before_cycle - 1;
            // 이를 사이클 길이 나머지 값을 취하면 유효한 값이 나온다.
            q %= cycle_length;
            cout << vv[before_cycle + 1 + q] << endl;
        }
    }
}