놀이공원 인형 맞추기

놀이공원에 가면 인형 맞추기 게임이 있다.

한 놀이공원에는 인형에 숫자를 써놓고 인형 맞추기를 한다.

그런데 맞춘 인형의 숫자의 합이 특정한 값이 되는 경우에만 맞춘 인형을 가져 갈 수 있다.

가령 10개의 인형에 쓰여진 숫자가 각각 25 27 3 12 6 15 9 30 21 19이고 숫자의 합이 50이 되는 경우만 인형을 가져갈수 있다면 6 19 25가 쓰여진 인형을 맞춰야 인형을 가져 갈 수 있다.

이때 인형의 갯수와 각각의 숫자와 필요한 합을 입력받으면 맞춰야 하는 인형의 숫자를 출력하는 프로그램을 작성해 보자.

입력 - 첫줄에는 인형의 갯수와 필요한 합, 둘째줄에는 인형 각각에 쓰여진 숫자

예)

10 50

25 27 3 12 6 15 9 30 21 19

출력 - 필요한 합이 되는 인형 각각의 숫자를 오름차순으로 출력

예)

6 19 25

※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.

6개의 풀이가 있습니다.

while __name__ == '__main__':
    res = []
    for s in ['a, b', 'c']:
        exec(s + ''' = [*map(int, input('>>>').split())]''')
    _ = [*([*(res.append(j) for j in __import__('itertools').combinations(c, i+1) if sum(j) == b)] for i in range(a))]
    for i in res:
        print(i)

속도는 고려하지 않았습니다. 10개 수준에서는 즉시 나옵니다. 파이썬 3.5.2 64

멋진 풀이입니다^^ list(map(...)) == [*map(...)] 인줄 이제 알았네요~ 컴프리헨션안에 append 사용하는것도 _ 변수로 처리하는것도 인상적이네요 - 디디, 2016/10/27 11:24 M D
+1 while __name__ == '__main__': for s in ['a, b', 'c']: exec(s + ''' = [*map(int, input('>>>').split())]''') for res in (j for i in range(a) for j in __import__('itertools').combinations(c, i+1) if sum(j) == b): print(sorted(res)) 그런데, append 하지 않아도 이렇게 하는것이 더 자연스러워 보여 올려봤습니다. Flair sizz님 풀이 공부하면서요. - 디디, 2016/10/28 11:47 M D
※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.
len_, sum_ = map(int, input(":").split(' '))
list_ = list(map(int, input(":").split(' ')))
result = [sorted(x) for x in [[list_[i] for i in range(len(list_)) if j & (1 << i)]
                              for j in range(1, 1 << len(list_))]
              if sum(x) == sum_]
print('\n'.join([','.join(map(str, x)) for x in result]))

Python 3.5.2에서 작성하였습니다.
bit연산으로 부분집합을 구해서 풀었습니다.

와 정말 뭐가뭔지 모르겠어요. 혹시 여유가 되시면 bit 연산을 어떻게 응용했는지 설명좀 부탁드립니다. - Dr.Choi, 2017/01/10 00:48 M D
전체 부분 집합을 만드는 방법이 진짜 신박한데 너무 궁금해요!!!! - Dr.Choi, 2017/01/10 01:55 M D
무식하게 풀었는데 감사합니다..^^;; 배열 수 만큼의 길이의 bit를 나열하고 1 이되는 자리수의 값을 배열로 묶어서 부분집합을 만들었습니다. [0, 1, 2] 1 : 0 0 1 : [2] 2 : 0 1 0 : [1] 3 : 0 1 1 : [1, 2] 4 : 1 0 0 : [0] 5 : 1 0 1 : [0, 2] 6 : 1 1 0 : [0, 1] 7 : 1 1 1 : [0, 1, 2] - Yeo HyungGoo, 2017/01/10 09:41 M D
※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.

파이썬3.5

죄송합니다... 위 댓글에는 지적질을 했는데 정작 제풀이는 형편이 없습니다.
속도도 최악입니다. 어떻게 손을 써볼수가 없었습니다.
입력, 출력 풀어조건에도 맞질 않습니다. 속도가 느리니 즐겁지가 않아 여기까지만 했습니다.

def f(a, target):

    a = sorted(a)
    result = set()

    def g(a):

        if sum(a) == target:
            result.add(tuple(a))
            return

        elif sum(a) < target:
            return

        else:
            for i in range(len(a)):
                x = a[:i] + a[i+1:]
                g(x)

    g(a)

    return result


a = [25, 27, 3, 12, 6, 15, 9, 30, 21, 19]
target = 50
f(a, target)

파이썬3.4.3

위의 방식보다 획기적으로 속도개선을 했습니다. 그러나 느리지요. 왜냐하면 완전탐색이니까요ㅜㅜ
보다 나은 알고리즘을 공부하면 다시 올리겠습니다.

# 제출 코드

# 입력
n, goal = tuple(map(int, input().split()))
a = sorted(list(map(int, input().split())))
assert len(a) == n

# 처리
result = []
def shot(last=0, picked=None, score=0):

    global result

    if picked is None:
        picked = []

    # 기저사례
    if score == goal:
        return picked

    elif score > goal:
        return False

    for i in range(last, n):      
        picked.append(a[i])
        ret = shot(i+1, picked, score+a[i])
        if ret: result.append(ret[:])    # result.append(ret)는 빈값이 됨.
        picked.pop()

# 출력
shot()
for line in result:
    for case in line:
        print(case, end=' ')
    print()
※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.

subsquence를 뽑아낼 때 재귀를 사용했습니다. 파이썬 3을 사용하였습니다.

def combination(x):
    if len(x)==1:
        return [[x[0]]]
    else:
        smallercombi=combination(x[:len(x)-1])
        realcombi=smallercombi[:]
        for smaller in smallercombi:
            realcombi.append(smaller+[x[len(x)-1]])
        realcombi.append([x[len(x)-1]])
        return realcombi

numofdoll, target =  map(int, input("두 수를 공백으로 구분하여 입력하시오 : ").split())
numlist = []
for x in range(numofdoll):
    numlist.append(int(input("인형에 써진 수를 입력하시오 : ")))
subsequence = combination(numlist)
for sub in subsequence:
    if sum(sub) == target:
        sub.sort()
        print(sub)
※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.
def sublist_(a):
    sublist=list()
    result=[]
    for i in range(1,1<<len(a)):
        k=str(bin(i))
        r=0
        for j in reversed(k[2:]):
            if j=="1":
               sublist.append(a[r])
            r+=1
        result.append(sublist)
        sublist=[]
    return result
a,b=map(int,input().split(" "))
setlist=list(map(int,input().split(" ")))
setlist_=sublist_(setlist)
for i in setlist_:
    if sum(i)==a:
        print(i,end=" ")

전체 부분 집합을 만들고 합을 구해서 같으면 출력합니다.

※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.
def doll(data):
    n,limit=data[0]
    data=sorted(data[1])
    if sum(data)<limit or min(data)>limit:
        return -1
    for i in range(n):
        if data[i]==limit:
            return [data[i]]
        elif doll([[n-1,limit-data[i]],data[0:i]+data[i+1:]])==-1:
            pass
        else:
            return [data[i]]+doll([[n-1,limit-data[i]],data[0:i]+data[i+1:]])
    return -1
doll([[10,50],[25, 27, 3, 12, 6, 15, 9, 30, 21, 19]])
※ 상대에게 상처를 주기보다 서로에게 도움이 될 수 있는 댓글을 달아 주세요.

풀이 작성

※ 풀이작성 안내
  • 본문에 코드를 삽입할 경우 에디터 우측 상단의 "코드삽입" 버튼을 이용 해 주세요.
  • 마크다운 문법으로 본문을 작성 해 주세요.
  • 풀이를 읽는 사람들을 위하여 풀이에 대한 설명도 부탁드려요. (아이디어나 사용한 알고리즘 또는 참고한 자료등)
  • 작성한 풀이는 다른 사람(빨간띠 이상)에 의해서 내용이 개선될 수 있습니다.
목록으로
코딩도장

코딩도장은 프로그래밍 문제풀이를 통해서 코딩 실력을 수련(Practice)하는 곳입니다.


언어별 풀이 현황
전 체 x 15
java x 4
python x 6
ruby x 1
기 타 x 2
cs x 1
matlab x 1