(BOJ) 15975 화살표 그리기

문제: https://www.acmicpc.net/problem/15975

 

문제 풀이:

https://hororolol.tistory.com/140 이 문제에서 long long으로 바꿔주고 

색이 하나 일 수도 있으므로 하나일 때는 연산 스킵해주면 됨

 

소스 코드:

typedef long long ll;
int n;
ll res;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL), cout.tie(NULL);

	unordered_map<ll, vector<ll>> m;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int q, w;
		cin >> q >> w;
		m[w].push_back(q);
	}
	for (auto next : m) {
		vector<ll> v = next.second;
		int len = v.size();
		if (len == 1) continue;
		sort(v.begin(), v.end());
		res += v[1] - v[0];
		for (int i = 1; i < len - 1; i++) {
			res += min(v[i + 1] - v[i], v[i] - v[i - 1]);
		}
		res += v[len - 1] - v[len - 2];
	}
	cout << res << '\n';
	return 0;
}