상자 꾸미기

문제

면이 6개인 상자가 있다. 이를 여러 가지 색종이를 붙여 꾸밀려고 하는데, 단 조건이 있다. 인접한 면에 같은 색의 색종이를 붙이면 안 된다는 것이다. 또한, 한 면에는 한 장의 색종이만 붙일 수 있다. 여러 가지 색의 색종이들이 주어졌을 때, 조건을 만족하여 상자의 모든 면에 붙일 수 있는지 판별하는 프로그램을 작성하시오.

 

입력

첫째 줄에 색종이의 장수 N ( 1 <= N <= 1,000 ) 이 주어진다. 둘째 줄에 각각의 색종이의 색깔을 나타내는 N개의 숫자가 주어진다. 색깔은 양의 정수로 이루어져 있고, 1부터 N까지의 범위의 수이다.

 

출력

조건을 만족하면서 상자를 꾸밀 수 있으면 “YES”, 아니면 “NO”를 출력한다.

 

 

예제 입력 

6 1 2 1 2 1 3

예제 출력

NO

 

예제 입력 

6 1 2 3 1 2 3

예제 출력

YES

 

예제 입력

7 1 1 1 2 2 3 3

예제 출력

YES

 

예제 입력

8 1 2 2 2 1 1 1 3

예제 출력

NO

 

#처음 접근법

distinctColor = list()
distinctColor = {}
isSuccess = True
 
for color in getColor:
  try: distinctColor[color]+= 1
  except: distinctColor[color]=1
 
 
 
if len(distinctColor) == 3:
  print(distinctColor)
  for k, v in distinctColor.items():
      if v != 2:  
        isSuccess = False
        break
elif len(distinctColor) == 4:
  pass
elif len(distinctColor) == 5:
  pass
elif len(distinctColor) > 5:
  pass
 
if isSuccess == True:
  print('YES')
else:
  print('NO')

# 개선된 접근법

N = int(input())
getColor = list(map(int, input().split()))
distinctColor = {} 
cnt = 0
 
for color in getColor:
  try: distinctColor[color]+= 1
  except: distinctColor[color]=1
 
 
for k, v in distinctColor.items():
  if v > 2:  
      cnt += 2
  else:
      cnt += v
  if cnt >= 6:
    break
 
 
if cnt >= 6:
  print('YES')
else:
  print('NO')

파이썬 중복제거 참조 aceatom.tistory.com/128

[python] 리스트 안에 있는 중복된 값 개수 구하기

#복습

import sys
 
n = int(sys.stdin.readline())
getInfo = list(map(int, sys.stdin.readline().split()))
getInfo_dic = {} 
 
for i in range(n):
  try: getInfo_dic[getInfo[i]] += 1
  except: getInfo_dic[getInfo[i]] = 1
 
if len(getInfo_dic) < 3:
  print('NO')
elif len(getInfo_dic) >= 3 and len(getInfo_dic) < 6:
  result = 0
  for k, v in getInfo_dic.items():
    if v > 2:
      result += 2
    else:
      result += v
  if result >= 6:
    print('YES')
  else:
    print('NO')
else:
  print('YES')