0205. [BOJ#28445] 알록달록 앵무새

내 코드

color_1, color_2 = input().split()
color_3, color_4 = input().split()

s = set([color_1, color_2, color_3, color_4])
lst = list(s)
lst.sort()

for i in lst:
    for j in lst:
        print(i, j)

set함수를 설정하는법을 몰라서 조금 헤맸는데 

set01 = {}  
set02 = set([iterable]) 

초기값으로 {}를 사용하면 딕셔너리가 되기 때문에 set()함수를 사용해서 list를 파라미터로 넘겨야 한다.

나머지는 리스트로 정렬 후 2차원 배열로 같은 배열을 사용하여 배열해주었다

 

1 초

1024 MB

611

465

412

81.262%

문제

재현이가 키우는 앵무새 포포와 레몬이는 그동안 새끼들을 참 많이도 낳았다. 그렇게 태어난 앵무새들을 관찰하며 재현이는 앵무새들의 색에 간단한 규칙이 있다는 것을 발견했다. 그것은 바로 자식 앵무새의 몸통 색은 아빠 새의 몸통 색과 꼬리 색, 엄마 새의 몸통 색과 꼬리 색 중 하나이며 꼬리 색도 마찬가지로 이 넷 중 하나의 색으로 정해진다는 것이다.

아빠 새의 몸통 색과 꼬리 색, 엄마 새의 몸통 색과 꼬리 색이 주어질 때 가능한 자식 앵무새의 몸통 색과 꼬리색의 모든 쌍을 사전 순으로 출력하라. (단, 중복되는 몸통 색, 꼬리 색의 쌍은 출력하지 않는다.)

입력

첫 번째 줄에 아빠 새의 몸통 색과 꼬리 색이 주어진다.

두 번째 줄에 엄마 새의 몸통 색과 꼬리 색이 주어진다.

각각의 색은 1$1$자 이상 20$20$자 이내의 알파벳 소문자로 이루어진 문자열로 주어지며, 같은 색이 중복되어 나타날 수 있다.

출력

자식 새의 몸통 색과 꼬리 색의 쌍을 한 줄에 하나씩 사전 순으로 출력한다. 사전 순으로 출력하라는 말은 몸통 색이 다르다면 몸통 색의 사전 순으로, 몸통 색이 같다면 꼬리 색의 사전 순으로 출력하라는 것을 의미한다.