[백준 / BOJ] 3003번 킹, 퀸, 룩, 비숍, 나이트, 폰 (C++, Python)

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

3003번: 킹, 퀸, 룩, 비숍, 나이트, 폰

---

문제

---

문제 풀이

for문을 활용하는 간단한 문제다.

 

 

C++ 상세 풀이

vector을 사용했다. 아래 코드처럼 입력받자마자 바로 출력해도 정답처리된다.

 

for (int i = 0; i < 6; ++i) {
    std::cin >> var[i];
    std::cout << chess[i] - var[i] << " ";
}

 

Python 상세 풀이

enumerate을 활용해 코드를 짧게 만들어보았다.

 

for i, v in enumerate(map(int, input().split())):
    print(chess[i] - v, end=" ")

 

 

코드 전문

C++

	#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 chess[6] = {1, 1, 2, 2, 2, 8};
	std::vector<int> var(6);
	for (int i = 0; i < 6; ++i) {
	std::cin >> var[i];
	std::cout << chess[i] - var[i] << " ";
	}
	return 0;
	}

view raw 3003_킹, 퀸, 룩, 비숍, 나이트, 폰 (22.09.03 성공).cpp hosted with ❤ by GitHub

 

Python

	import sys
	input = sys.stdin.readline
	if __name__ == "__main__":
	chess = [1, 1, 2, 2, 2, 8]
	for i, v in enumerate(map(int, input().split())):
	print(chess[i] - v, end=" ")

view raw 3003_킹, 퀸, 룩, 비숍, 나이트, 폰 (22.09.03 성공).py hosted with ❤ by GitHub

 

소감

enumerate를 잘 활용하면 굳이 for i in range()문을 사용하지 않아도 된다.