21-fnhid

[Fnhid]

🔗 문제 링크

좋다

✔️ 소요된 시간

1시간
테스트케이스가 생각보다 빡빡했습니다 ..
예외처리 생각 잘 해줘야함..

✨ 수도 코드

N개의 수 중에서 다른 수 두 개의 합으로 이루어진 '좋은 수'의 개수를 찾는 문제입니다.

저는 투 포인터로 해결했습니다.

4
1 2 3 4 

위 케이스에서 3이 '좋은 수'인지 검사해 봅시다.

target<sum이므로 R--해줍니다.

target 포인터와 R 포인터가 같은 곳을 가리키고 있습니다. '다른 두 수의 합'을 구해야 하므로 R--해줍니다.

1+2=3임을 찾았습니다. 3은 '좋은 수'입니다.

이런 식으로 모든 수에 이 과정을 반복해주면 좋은 수의 개수를 구할 수 있습니다.

수도 코드

정수 n을 입력받는다
n개의 정수를 입력받아 리스트 v에 저장한다

v를 오름차순으로 정렬한다

좋은 수의 개수를 저장할 변수 cnt를 0으로 초기화한다

i를 0부터 n-1까지 반복하면서 다음을 수행한다:
    target에 v[i] 값을 저장한다
    왼쪽 포인터 l을 0으로, 오른쪽 포인터 r을 n-1로 초기화한다

    l이 r보다 작을 동안 반복한다:
        만약 l이 i와 같다면:
            l을 1 증가시키고 다음 반복으로 넘어간다

        만약 r이 i와 같다면:
            r을 1 감소시키고 다음 반복으로 넘어간다

        v[l]과 v[r]의 합을 sum에 저장한다

        만약 sum이 target과 같다면:
            cnt를 1 증가시키고 반복문을 종료한다

        만약 sum이 target보다 작다면:
            l을 1 증가시킨다

        그렇지 않다면:
            r을 1 감소시킨다

cnt를 출력한다

📚 새롭게 알게된 내용

[froglike6]

저는 이 문제를 2년 전에 투 포인터를 이용해 풀었습니다(좋다). 기억이 잘 나진 않지만, 다시 복습을 하며 코드를 보니 유현님의 설명과 매우 흡사한 것 같습니다. 파이썬이다보니 확실히 편한 것 같습니다.

import sys

N = int(sys.stdin.readline())
arr = list(map(int, sys.stdin.readline().split()))
arr.sort()
ans = 0

for i in range(N):
    tmp = arr[:i] + arr[i + 1:]
    left, right = 0, len(tmp) - 1
    while left < right:
        t = tmp[left] + tmp[right]
        if t == arr[i]:
            ans += 1
            break
        if t < arr[i]: left += 1
        else: right -= 1
print(ans)

[dohyeondol1]

처음에 이분탐색으로 생각했다가 투포인터로 틀어서 맞췄습니다.
아무래도 코드만 놓고 보면 이분탐색 코드랑 비슷한 부분이 많은 것 같네요..~!

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int A[2001];

int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    
    int N, answer = 0, target;
    cin >> N;
    for(int i = 0; i < N; i++)
        cin >> A[i];
    sort(A, A + N);

    for(int i = 0; i < N; i++){
        target = A[i];
        int left = 0, right = N-1, sum;
        while(left < right){
            sum = A[left] + A[right];
            if(sum == target){ 
                if(left!= i && right != i){
                    answer++;
                    break;
                }
                else if(left== i) left++;
                else if(right == i) right--;
            }
            else if(sum < target) left++;
            else right--;
        }
    }
    cout << answer;
    return 0;
}