[BOJ 30814 // C++] Watchmen

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 30814번 문제인 Watchmen이다.
문제는 아래 링크를 확인하자.

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

 

"두 점 사이의 Manhattan Distance와 Euclidean Distance가 같다"는 것은 "두 점의 \(x\)좌표가 서로 같거나 \(y\)좌표가 서로 같다"는 것과 동치임을 어렵지 않게 증명할 수 있다.

 

이를 이용하면, 문제에서 구하는 값은 \(x\)좌표 또는 \(y\)좌표가 서로 일치하는 점의 쌍의 수를 세는 것으로 해결할 수 있다. 글쓴이는 앞서 나온 점들 중 \(x\)좌표가 같은 점의 개수와 \(y\)좌표가 같은 점의 개수를 각각 기록해 문제를 해결했다. 이 과정에서 좌표가 일치하는 두 점의 경우 두 경우 모두에 포함되어 두 번씩 계수되므로 이에 유의하여 구현하자.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <map>
using namespace std;
typedef long long ll;

int N;
map<int, int> mpx, mpy;
map<pair<int, int>, int> mp;
ll ans;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N;
	while (N--) {
		int x, y; cin >> x >> y;
		ans += mpx[x];
		ans += mpy[y];
		ans -= mp[make_pair(x, y)];
		mpx[x]++, mpy[y]++, mp[make_pair(x, y)]++;
	}
	cout << ans;
}