[백준/파이썬&자바] #2470

배열을 정렬하고 투 포인터를 쓰면 되겠구나 싶던 문제였다. 

 

나름 확신하고 문제를 풀었는데 처음에는 답이 틀렸다고 나왔다. 입력을 고려해 보니 엣지 케이스로는 1,000,000,000이 100,000개 있는 경우가 주어질 수 있겠다는 생각이 들었다. 

 

기존에는 두 용액의 합 중 가장 0에 가까운 값을 저장하는 변수(zero_sum)를 1e9로 초기화했는데, 이러면 이 엣지 케이스의 값을 반영하지 못하겠다는 생각이 들었다. 

 

해당 값만 sys.maxsize로 바꾸어 주니 잘 동작하였다:)

 

파이썬 풀이

import sys

input = sys.stdin.readline

N = int(input())
arr = list(map(int, input().split()))
arr.sort()

zero_sum = sys.maxsize
left, right = 0, N-1
zero_left, zero_right = 0, N-1
while left < right:
    current_sum = arr[left] + arr[right]
    if abs(current_sum) < abs(zero_sum):
        zero_left, zero_right = arr[left], arr[right]
        zero_sum = current_sum
        if current_sum == 0:
            break
    else:
        if current_sum > 0:
            right -= 1
        else:
            left += 1

print(f"{zero_left} {zero_right}")

 

자바 풀이

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        long[] arr = new long[N];
        for(int i=0; i<N; i++) {
            arr[i] = Long.parseLong(st.nextToken());
        }
        br.close();

        Arrays.sort(arr);

        int left = 0;
        int right = N-1;

        long answerLeft = 0L;
        long answerRight = 0L;
        long answer = Long.MAX_VALUE;
        long currentSum = 0L;

        while(left < right) {
            currentSum = arr[left] + arr[right];
            if(Math.abs(currentSum) < Math.abs(answer)) {
                answer = currentSum;
                answerLeft = arr[left];
                answerRight = arr[right];
                if(currentSum == 0) {
                    break;
                }
            }

            if (currentSum > 0) {
                right--;
            } else {
                left++;
            }
        }   
        System.out.println(answerLeft + " " + answerRight);
    }
}

궁금증

• 왜 파이썬과 달리 자바는 byte, short, int, long으로 나눠져 있을까?