BOJ_1253 : 좋다

👉 문제링크

1253번: 좋다

---

---

🔸 문제 분석 🔸

• N개의 수에서 서로 다른 두 수로 만들 수 있는 수의 개수를 출력한다.
• 인덱스가 다르면 값이 같아도 다른 수이다.
• 시간 복잡도
  • N의 개수가 최대값 2,000이라 가정해도 좋은 수 하나를 찾는 알고리즘의 시간 복잡도는 N^2보다 작아야 한다.
    (N^2 를 사용하면 다시 N번 반복해서 최종 시간 복잡도는 N^3가 되기 때문)
  • 좋은 수 하나를 찾는 알고리즘은 최소 O(nlongn)이어야 한다. 정렬과 투 포인터 알고리즘 사용
    • 단, 정렬된 데이터에서 자기 자신을 좋은 수 만들기에 포함하지 않도록 주의

🔸 코드 🔸

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        long arr[] = new long[n];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++)
            arr[i] = Long.parseLong(st.nextToken());

        Arrays.sort(arr);

        int answer = 0;
        for (int k = 0; k < n; k++){ // arr[k]가 좋은 수인지 확인
            long find = arr[k];
            int i = 0;
            int j = n - 1;
            while(i < j){   // 투 포인터
                if (arr[i] + arr[j] == find){
                    if (i != k && j != k){ // 자기 자신은 제외하고 서로 다른 수여야 함(위치만 달라도 됨)
                        answer++;
                        break;
                    } else if (i == k) i++;
                    else if (j == k) j--;
                } else if (arr[i] + arr[j] < find) i++;
                else j--;
            }
        }
        System.out.println(answer);
        br.close();
    }
}

🔸 코드 해석 🔸

• N개의 수를 입력받아 정렬한다. N은 최대 10억이므로 N과 arr를 long형으로 선언한다.
• 투 포인터 i, j를 배열 arr 양쪽 끝에 위치시키고 투 포인터 이동 원칙을 활용해 탐색한다.
  • A[i] + A[j] > K : j--;
  • A[i] + A[j] < K : i++;
  • A[i] + A[j] == K : i++; j--; count++;
• arr의 인덱스 0부터 N-1 까지 투 포인터로 탐색한다.
  • 투 포인터가 자기자신(K) 가 되지 않도록 주의한다.

---

🔸 end 🔸

• 시간 복잡도도 계산이 안되고 투 포인터를 어떻게 적용해야할지 감이 안잡혀서 교재를 보면서 풀이했다.
  • 시간 복잡도는 n의 최대값에 연산할 알고리즘의 시간 복잡도를 곱해서 1억이 1초일때 넘어가는지 확인해야한다.
  • 투 포인터는 아직 자유롭게 다루지 못하는 것 같아서 더 연습이 필요해 보인다.
• 태그를 보니 이분 탐색이나 해시를 이용해서도 풀이가 되는 것 같다.