본문 바로가기
반응형

itertools4

[백준][Python] 16439번 치킨치킨치킨 16439번: 치킨치킨치킨 첫 번째 줄에 고리 회원의 수 N (1 ≤ N ≤ 30) 과 치킨 종류의 수 M (3 ≤ M ≤ 30) 이 주어집니다. 두 번째 줄부터 N개의 줄에 각 회원의 치킨 선호도가 주어집니다. i+1번째 줄에는 i번째 회원의 선 www.acmicpc.net 코드 from itertools import combinations import sys n,m = map(int,input().split()) a_list = [] max_t = 0 for _ in range(n): a_list.append(list(map(int,sys.stdin.readline().split()))) comb = list(combinations(range(m),3)) for a,b,c in comb: total .. 2022. 7. 22.
[백준][Python] 5568번 카드 놓기 5568번: 카드 놓기 예제 1의 경우 상근이는 11, 12, 21, 112, 121, 122, 212를 만들 수 있다. www.acmicpc.net 코드 1 from itertools import permutations n = int(input()) k = int(input()) n_list = [] dic = {} cnt = 0 for _ in range(n): n_list.append(int(input())) m_list = list(permutations(n_list,k)) for a in m_list: b = '' for i in a: b += str(i) if b not in dic: dic[b] = 1 cnt+=1 print(cnt) permutations 함수를 이용해서 만들 수 있는 모든.. 2022. 7. 8.
[백준][Python] 2503번 숫자 야구 2503번: 숫자 야구 첫째 줄에는 민혁이가 영수에게 몇 번이나 질문을 했는지를 나타내는 1 이상 100 이하의 자연수 N이 주어진다. 이어지는 N개의 줄에는 각 줄마다 민혁이가 질문한 세 자리 수와 영수가 답한 스트 www.acmicpc.net 코드 import sys from itertools import permutations n = int(input()) n_list = [] for i in range(n): a = list(map(int,sys.stdin.readline().split())) n_list.append(a) items = [1,2,3,4,5,6,7,8,9] i_list = list(permutations(items,3)) cnt = 0 for a in i_list: flag = 0.. 2022. 7. 7.
[백준][Python] 18511번 큰 수 구성하기 18511번: 큰 수 구성하기 첫째 줄에 N, K의 원소의 개수가 공백을 기준으로 구분되어 자연수로 주어진다. (10 ≤ N ≤ 100,000,000, 1 ≤ K의 원소의 개수 ≤ 3) 둘째 줄에 K의 원소들이 공백을 기준으로 구분되어 주어진다. 각 www.acmicpc.net 코드 from itertools import product n, m = map(int,input().split()) k = list(map(int,input().split())) k.sort(reverse=True) length = len(str(n)) while 1: num = list(product(k, repeat=length)) for a in num: a_num = int(''.join(map(str,a))) if a_num 2022. 7. 7.
반응형

티스토리툴바