백준 1253번 좋다

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

1253번: 좋다

해쉬 맵 설계를 잘해야하는 문제였습니다.

좋은 숫자의 정의는 "해당 숫자를 제외한 다른 두 숫자의 합으로 해당 숫자를 표현할 수 있는 숫자" 이므로

해당 숫자의 좋은 숫자 여부와, 인덱스를 저장해야 합니다.

	#include <iostream>
	#include <vector>
	#include <map>
	using namespace std;
	int main(void)
	{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int N;
	cin >> N;
	map<int, pair<bool, int>> visited; // {숫자, {좋은 수 여부, 현재 숫자 인덱스}}
	vector<int> v(N);
	for (int i = 0; i < N; i++)
	{
	cin >> v[i];
	visited[v[i]] = { false, i };
	}
	for (int i = 0; i < N; i++)
	{
	for (int j = i + 1; j < N; j++)
	{
	int sum = v[i] + v[j];
	if (visited.count(sum))
	{
	// 반드시 다른 숫자의 합
	if (visited[sum].second == i || visited[sum].second == j)
	{
	continue;
	}
	visited[sum].first = true;
	}
	}
	}
	int result = 0;
	for (int i = 0; i < N; i++)
	{
	if (visited[v[i]].first == true)
	{
	result++;
	}
	}
	cout << result << "\n";
	return 0;
	}

view raw .cpp hosted with ❤ by GitHub

개발환경:Visual Studio 2017

 

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