[백준 / BOJ] 10816번 숫자 카드 2 (C++, Python)

링크 : https://www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

---

문제

---

문제 풀이

숫자 카드와 유사한 문제이지만 이번에는 1, 0만 출력하는 게 아니라 몇 개를 가지고 있는지 출력해야 한다.

upper_bound와 lower_bound 반복자의 차이를 구하면 그것이 해당 숫자 카드의 갯수다.

 

파이썬은 언제나 그랬듯이 간단하게 풀린다. 인터넷에 다른 풀이들도 많겠지만 Collections 모듈을 활용해도 된다.

 

 

C++ 코드 전문

	#include <algorithm>
	#include <iostream>
	#include <vector>
	void init() {
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	}
	int main() {
	init();
	int N, M;
	std::cin >> N;
	std::vector<int> v1(N);
	for (int i = 0; i < N; ++i)
	std::cin >> v1[i];
	std::cin >> M;
	std::vector<int> v2(M);
	for (int i = 0; i < M; ++i)
	std::cin >> v2[i];
	std::sort(v1.begin(), v1.end());
	for (int i = 0; i < M; ++i) {
	if (std::binary_search(v1.begin(), v1.end(), v2[i])) {
	std::cout << std::upper_bound(v1.begin(), v1.end(), v2[i]) - std::lower_bound(v1.begin(), v1.end(), v2[i]) << " ";
	}
	else
	std::cout << "0 ";
	}
	return 0;
	}

view raw 10816_숫자 카드 2 (22.05.14 성공).cpp hosted with ❤ by GitHub

 

Python 코드 전문

	import sys
	from collections import Counter
	input = sys.stdin.readline
	N = input()
	arr = input().split()
	M = input()
	ans = input().split()
	arr = Counter(arr)
	print(' '.join(f'{arr[i]}' if i in arr else '0' for i in ans))

view raw 10816_숫자 카드 2 (22.05.14 성공).py hosted with ❤ by GitHub

 

소감