[BOJ][C++] 백준 3273번 : 두 수의 합

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

3273번: 두 수의 합

 

문제

n개의 서로 다른 양의 정수 a1, a2, ..., an으로 이루어진 수열이 있다. ai의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, ai + aj = x (1 ≤ i < j ≤ n)을 만족하는 (ai, aj)쌍의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)

출력

문제의 조건을 만족하는 쌍의 개수를 출력한다.

 

풀이1 - 시간 초과

#include <iostream>
#include <vector>

using namespace std;

int main() {
    int n,x,ans=0;
    vector <int> v;
    
    cin >> n;
    for(int i=0, tmp; i<n; i++){
        cin >> tmp;
        v.push_back(tmp);
    }
    cin >> x;
    
    for(int i=0; i<n; i++){
        int tmp = x - v[i];
        for(int i=0; i<n; i++){
            if(tmp == v[i]) {
                ans++;
                break;
            }
        }
    }
    
    cout << ans/2;  //중복제거
    
    return 0;
}

답은 맞게 나오는 것 같은데 벡터를 이중으로 돌아야해서 시간복잡도가 불안불안하긴 했다...

역시.. 아주 화려하게 시간초과

 

풀이2 - 런타임에러

에휴.. 그래서 for문(시간복잡도 O(n))이 아닌 그냥 바로 (O(1)) x-원소 라는 숫자가 있는지 판단할 수 있도록 숫자 여부를 판단하는 bool 배열을 만들어줬다.

처음에는 int 로 했는데 자꾸 OutOfBounds 가 떠서 정수배열은 1000000 최대인거때문인가 싶어 열심히 바꾸고 삽질함..

#include <iostream>
#include <vector>

using namespace std;

bool a[1000001];

int main() {
    int n,x,ans=0;
    vector <int> v;
    
    cin >> n;
    for(int i=0, tmp; i<n; i++){
        cin >> tmp;
        a[tmp] = true;
        v.push_back(tmp);
    }
    
    cin >> x;
    
    for(int i=0; i<n; i++){
        int tmp = x - v[i];
        if (a[tmp])
            ans++;
    }
    
    cout << ans/2;  //중복제거
    
    return 0;
}

무수한 삽질의 흔적

아 근데 알고보니 전혀 다른곳에서 문제가 생긴거였다

 

풀이3 - 성공

위 코드는 그대로 쓰고 중간에 x-원소 인 tmp 를 배열에 넣어서 처리하기 전 필터링을 했다

문제를 잘 읽자..

보면 원소는 1~1000000 이고

x는 1~2000000이다

 

그러니까.. x에서 원소를 빼면 2000000-1 ~ 1-1000000 이렇게 험난한 범위가 생긴다

음수도 있고 배열크기 넘어서는 애도 있으니 런타임에러가 7프로부터 나는거였다.. 아주 일정하게..

 

처음에는 음수만 예외처리했는데 삽질끝에 1000000이하의 tmp도 처리해줌

#include <iostream>
#include <vector>

using namespace std;

bool a[1000001];

int main() {
    int n,x,ans=0;
    vector <int> v;
    
    cin >> n;
    for(int i=0, tmp; i<n; i++){
        cin >> tmp;
        a[tmp] = true;
        v.push_back(tmp);
    }
    
    cin >> x;
    
    for(int i=0; i<n; i++){
        int tmp = x - v[i];
        if (tmp > 0 && tmp < 1000001 && a[tmp]) //여기 조건을 두개 추가했다
            ans++;
    }
    
    cout << ans/2;  //중복제거
    
    return 0;
}

 

TIL

문제를 잘 읽자 ^^