<Hackerrank 문제풀이: 파이썬> List Comprehensions

 

간단하지만 매우 긴 문제였다:

if __name__ == '__main__':
    x = int(input())
    y = int(input())
    z = int(input())
    n = int(input())
    list_of_permutations_before_delete = []
    list_of_permutations = []
    for i in range(x+1):
        for j in range(y+1):
            for k in range(z+1):
                permutations = [0, 0, 0]
                permutations[0] = i
                permutations[1] = j
                permutations[2] = k
                list_of_permutations_before_delete.append(permutations)
    for l in list_of_permutations_before_delete:
        if sum(l) != n:
            list_of_permutations.append(l)
            
    print(list_of_permutations)

단순하게 리스트를 일일이 다 구현하여 비어 있는 리스트에 더해주고, n 값에 따라 출력해주고 출력해주지 않을 리스트를 나누어 놓은 리스트를 정의하고 출력해준다.