17827 -달팽이 리스트 (백준)

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

 

순환 리스트 만들어야 해서 노드 따로 만들고 하는건줄 알았는데, 시간초과가 많이 나서 잘못된걸 느끼고 수정했다. 

 

순환 길이를 계산하고  각 질문에 대한 값을 구하는게 오래걸렸다. 

 

항상 느끼지만 문제를 잘게잘게 나누고, 부분부분 구현하면서 차근차근 단계 밟아가는게 알고리즘 푸는데 핵심인거 같다.

#include <iostream>
#include <vector>

using namespace std;

int main() 
{
	int N, M, V;
	cin >> N >> M >> V;

	// 노드 값 입력
	vector<int> values(N + 1);
	for (int i = 1; i <= N; ++i) 
	{
		cin >> values[i];
	}

	// 질문 입력
	vector<long long> queries(M);
	for (int i = 0; i < M; ++i) 
	{
		cin >> queries[i];
	}

	// 순환 길이 계산
	int cycle_length = N - V + 1;

	// 각 질문 처리
	for (int i = 0; i < M; ++i) 
	{
		// 입력 값
		long long K = queries[i];
		int current_index;

		// 입력값이 순환 길이보다 작을때
		if (K < V - 1) 
		{
			current_index = 1 + K;
		}
		// 순환 이전의 값을 빼고 나머지 리스트에서만 순환
		else 
		{
			K -= (V - 1);
			current_index = V + (K % cycle_length);
		}

		cout << values[current_index] << "\n";
	}

	return 0;
}

 long long도 써야 하고 단위가 큰걸 고려했어야 했는데 자주 틀려서 고역이다.