자료구조 및 알고리즘/알고리즘

[알고리즘] 이것이 취업을 위한 코딩 테스트다 with 파이썬 - 볼링공 고르기

benjykim 2021. 1. 26. 17:07
반응형

이것이 취업을 위한 코딩 테스트다 with 파이썬 - 볼링공 고르기

  • 내 풀이

    from itertools import combinations
    import sys
    input = sys.stdin.readline
    
    n, m = map(int, input().split())
    bowling_balls = list(map(int, input().split()))
    count = 0
    
    for combi in combinations([i for i in range(1, n+1)], 2):
        if bowling_balls[combi[0]-1] != bowling_balls[combi[1]-1]:
            count += 1
    
    print(count)
반응형