백준 1253 좋다

문제

N개의 수 중에서 어떤 수가 다른 수 두 개의 합으로 나타낼 수 있다면 그 수를 “좋다(GOOD)”고 한다.

N개의 수가 주어지면 그 중에서 좋은 수의 개수는 몇 개인지 출력하라.

수의 위치가 다르면 값이 같아도 다른 수이다.

입력

첫째 줄에는 수의 개수 N(1 ≤ N ≤ 2,000), 두 번째 줄에는 i번째 수를 나타내는 Ai가 N개 주어진다. (|Ai| ≤ 1,000,000,000, Ai는 정수)

출력

좋은 수의 개수를 첫 번째 줄에 출력한다.

예제 입력 1 복사

10
1 2 3 4 5 6 7 8 9 10

예제 출력 1 복사

8

 

• 풀이

 

-실패한 풀이

 

n이 2000개니 n^2을 하고 set을 이용해서 풀어도 2초 이내는 넉넉할줄 알고 

n^2으로 순회하면서 set에 합을 저장해둬서 있는지 판단하는 식으로 코드를 짜고 실패했다..

 

public class Main {

    public static void main(String[] args) throws IOException {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(), lt = 0, rt = 0;
        long res = 0;
        long[] arr= new long[n];
        Set<Long> set = new HashSet<>(); // 원소 입력받는 set
        Set<Long> set2 = new HashSet<>(); // 합 저장해두는 set
        for (int i = 0; i < n; i++){
            arr[i] = in.nextLong();
            set.add(arr[i]);
        }

        for (int i = 0; i < n; i++){
            for (int j = i+1; j < n; j++){
                if (set.contains(arr[i]+arr[j]) && !set2.contains(arr[i]+arr[j])) {
                    set2.add(arr[i]+arr[j]);
                    res++;
                }
            }
        }
        System.out.println(res);
    }
}

 

- 투 포인터로 푸는 방법을 보고 풀어본 풀이 <- 46퍼에서 막힘.. 이유를 모르겠슴

다른 수 두 개의 합이므로 lt와 rt가 같아지지 않는 조건인 while(lt < rt) 에서 타겟 인덱스가 lt, rt인 경우를 제외하고 

arr[lt] + arr[rt]가 타겟보다 크면 rt--;

arr[lt] + arr[rt]가 타겟보다 작으면 lt++을 하는 투포인터 연산을 통해 답을 구한다 까지 보고 코드를 작성했다

46퍼에서 멈추는 이유가 스캐너를 써서 인가 하고 스캐너를 버퍼로 교체해봐도 계속 46퍼에서 틀렸지만 결국

이유를 찾아내진 못했다..

import java.util.*;
import java.io.*;

public class Main {

    static int N;
    static int[] A;

    static boolean solution(int idx){
        int lt = 1, rt = N;
        int target = A[idx];
        while (lt < rt){
            
            //타겟 인덱스랑 lt, rt가 같으면 제외하고 계산
            if (lt == idx) lt++;
            else if (rt == idx) rt--;

            if (lt >= rt) return false;

            if (A[lt] + A[rt] > target) rt--;
            else if (A[lt] + A[rt] == target) return true;
            else lt++;
        }
        return false;
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        int res =0 ;
        A = new int[N +1];
        String s = br.readLine();
        StringTokenizer st = new StringTokenizer(s);
        for (int i = 1; i <= N; i++) A[i] = Integer.parseInt(st.nextToken());
        Arrays.sort(A);

        for (int i = 1; i <= N; i++){
            if (solution(i)) {
                /*System.out.println(A[i]);*/
                res++;
            }
        }
        System.out.println(res);
    }
}

 

강의 정답 풀이

import java.io.*;
import java.util.*;

public class Main {
    static FastReader scan = new FastReader();
    static StringBuilder sb = new StringBuilder();

    static int N;
    static int[] A;

    static void input() {
        N = scan.nextInt();
        A = new int[N + 1];
        for (int i = 1; i <= N; i++) {
            A[i] = scan.nextInt();
        }
    }

    // target_idx 번째 원소가 서로 다른 두 수의 합으로 표현이 되는가?
    static boolean func(int target_idx) {
        int L = 1, R = N;
        int target = A[target_idx];
        while (L < R) {
            if (L == target_idx) L++;
            else if (R == target_idx) R--;
            else {
                if (A[L] + A[R] > target) R--;
                else if (A[L] + A[R] == target) return true;
                else L++;
            }
        }
        return false;
    }

    static void pro() {
        // 최소, 최대를 빠르게 알기 위한 정렬
        Arrays.sort(A, 1, N + 1);

        int ans = 0;
        for (int i = 1; i <= N; i++) {
            // i 번째 원소가 서로 다른 두 수의 합으로 표현이 되는가?
            if (func(i)) ans++;
        }
        System.out.println(ans);
    }

    public static void main(String[] args) {
        input();
        pro();
    }


    static class FastReader {
        BufferedReader br;
        StringTokenizer st;

        public FastReader() {
            br = new BufferedReader(new InputStreamReader(System.in));
        }

        public FastReader(String s) throws FileNotFoundException {
            br = new BufferedReader(new FileReader(new File(s)));
        }

        String next() {
            while (st == null || !st.hasMoreElements()) {
                try {
                    st = new StringTokenizer(br.readLine());
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return st.nextToken();
        }

        int nextInt() {
            return Integer.parseInt(next());
        }

        long nextLong() {
            return Long.parseLong(next());
        }

        double nextDouble() {
            return Double.parseDouble(next());
        }

        String nextLine() {
            String str = "";
            try {
                str = br.readLine();
            } catch (IOException e) {
                e.printStackTrace();
            }
            return str;
        }
    }
}