백준 15970번 화살표 그리기

문제 링크입니다: https://www.acmicpc.net/problem/15970

15970번: 화살표 그리기

각 색깔마다 좌표들을 저장하고 오름차순 정렬을 한 뒤 가장 가까운 점끼리 직선을 이은 거리들의 합을 구하면 되는 문제였습니다.

자세한 내용은 코드 주석을 참고해주세요.

 

	#include <iostream>
	#include <vector>
	#include <algorithm>
	using namespace std;
	const int MAX = 5000 + 1;
	vector<int> coords[MAX];
	int main(void)
	{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int N;
	cin >> N;
	for (int i = 0; i < N; i++)
	{
	int coord, color;
	cin >> coord >> color;
	coords[color].push_back(coord);
	}
	long long result = 0;
	for (int color = 1; color < MAX; color++)
	{
	// 해당 색깔이 없다면 pass
	if (coords[color].empty())
	{
	continue;
	}
	sort(coords[color].begin(), coords[color].end());
	// 최좌측 점은 바로 오른쪽에 있는 점과 이어지고
	result += coords[color][1] - coords[color][0];
	for (int j = 1; j < coords[color].size() - 1; j++)
	{
	int x = coords[color][j];
	// 양 옆 점 중 최소 거리를 결과에 더한다
	result += min(x - coords[color][j - 1], coords[color][j + 1] - x);
	}
	int lastIdx = coords[color].size() - 1;
	// 최우측 점은 바로 좌측에 있는 점과 이어집니다.
	result += coords[color][lastIdx] - coords[color][lastIdx - 1];
	}
	cout << result << "\n";
	return 0;
	}

view raw .cpp hosted with ❤ by GitHub

개발환경:Visual Studio 2017

 

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~