[BOJ/백준] 9024 두 수의 합 - 파이썬

문제

https://www.acmicpc.net/problem/9024

 

 

문제풀이

 

num 배열을 정렬하고 k값 보다 큰지 아닌지 비교해가면서 이분탐색을 진행한다.

그 값이 k보다 크면 mid-1 값을 r로 설정해주고

                   작으면 min+1 값을 l로 설정해준다.

k와 두 수의 합의 차이가 mini 값보다 작으면 mini 값을 없데이트 해주고 cnt=1로 초기화한다.

                                mini 값과 같으면 cnt+=1을 해준다.

 

 

코드

import sys
input=sys.stdin.readline

t=int(input())
for i in range(t):
    n,k=map(int,input().split())
    num=list(map(int,input().split()))
    num.sort()

    mini=200000000
    for j in range(n):
        l=j+1
        r= n-1
        while(l<=r):
            mid=(l+r)//2
            s=num[j]+num[mid]
            if(s>k):
                r=mid-1
            else:
                l=mid+1
            if(abs(k-s)<mini):
                cnt=1
                mini=abs(k-s)
            elif(abs(k-s)==mini):
                cnt+=1
    print(cnt)