본문 바로가기

알고리즘/해커랭크

Divisible Sum Pairs

문제

https://www.hackerrank.com/challenges/divisible-sum-pairs/problem?h_r=next-challenge&h_v=zen&h_r=next-challenge&h_v=zen&h_r=next-challenge&h_v=zen

 

리스트 ar과 k가 주어졌을 때 ar[i]+ar[j]%k=0인 i,j 쌍의 수를 출력 (이때, i < j )

입력

len(ar) k

ar

 

 

입력 예

6 3 

1 3 2 6 1 2

 

출력 예

5

 

ar[0]+ar[2]=3

ar[0]+ar[5]=3

ar[2]+ar[4]=3

ar[1]+ar[3]=9

ar[4]+ar[5]=3

은 모두 k=3으로 나누어 떨어짐.

 

제출한 코드

def divisibleSumPairs(n, k, ar):
    res=0
    for i, val1 in enumerate(ar):
        for j, val2 in enumerate(ar):
            if i>j and (val1+val2)%k==0:
                res+=1
    return res      

n^2 벗어나고 싶어서 고민해봤는데 딱히 다른 방법이 생각안난다..

'알고리즘 > 해커랭크' 카테고리의 다른 글

Day of Programer  (0) 2020.03.08
Migratory Birds  (0) 2020.03.08
Birthday Chocolate  (0) 2020.03.08
Breaking the Records  (0) 2020.03.07
Between Two Sets  (0) 2020.03.07