Do it! 알고리즘 코딩테스트 | 1. 자료구조 - 투 포인터 | 007 주몽의 명령

문제 정보 백준 1940번 주몽 https://www.acmicpc.net/problem/1940

 

문제

주몽은 철기군을 양성하기 위한 프로젝트에 나섰다. 그래서 야철대장을 통해 철기군이 입을 갑옷을 만들게 하였다. 야철대장은 주몽의 명에 따르기 위하여 연구에 착수하던 중 아래와 같은 사실을 발견하게 되었다.

갑옷을 만드는 재료들은 각각 고유한 번호를 가지고 있다. 갑옷은 두 개의 재료로 만드는데 두 재료의 고유한 번호를 합쳐서 M(1 ≤ M ≤ 10,000,000)이 되면 갑옷이 만들어 지게 된다. 야철대장은 자신이 만들고 있는 재료를 가지고 갑옷을 몇 개나 만들 수 있는지 궁금해졌다. 이러한 궁금증을 풀어 주기 위하여 N(1 ≤ N ≤ 15,000) 개의 재료와 M이 주어졌을 때 몇 개의 갑옷을 만들 수 있는지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 재료의 개수 N(1 ≤ N ≤ 15,000)이 주어진다. 그리고 두 번째 줄에는 갑옷을 만드는데 필요한 수 M(1 ≤ M ≤ 10,000,000) 주어진다. 그리고 마지막으로 셋째 줄에는 N개의 재료들이 가진 고유한 번호들이 공백을 사이에 두고 주어진다. 고유한 번호는 100,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 갑옷을 만들 수 있는 개수를 출력한다.

예제 입력 1 

6
9
2 7 4 1 5 3

예제 출력 1 

2

 

//내 코드
#include<iostream>
using namespace std;
int swap(int *x, int*y){
    int t = *x;
    *x = *y;
    *y = t;
}
int main(){
    int n, m, a[15000];
    cin >> n >> m;
    for(int i=0; i<n; i++){
        cin >> a[i];
    }
    for(int i=0; i<n; i++){
        for(int j=0; j<n-1; j++){
            if(a[j] > a[j+1]){
                swap(&a[j], &a[j+1]);
            }
        }
    }
    int start=0;
    int end=start+1;
    int cnt=0;
    while(start < n){
        if(a[start] + a[end] == m && (a[start] != 0 && a[end] != 0)){
            a[start] = 0;
            a[end] = 0;
            cnt++;
            start++;
            end++;
        }
        else if(a[start] + a[end] > m && (a[start] != 0 && a[end] != 0)){
            start++;
            end = start;
        }
        else{
            end++;
        }
        if(a[start] == 0){
            start++;
        }
        if(a[end] == 0){
            end++;
        }
    }
    cout << cnt;
}

 

//정답 코드
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int N, M;
    cin >> N >> M;
    vector<int> A(N, 0);
    for (int i = 0; i < N; i++) {
        cin >> A[i];
    }
    sort(A.begin(), A.end());

    int count = 0;
    int i = 0;
    int j = N - 1;
    while (i < j) { //투 포인터 이동 원칙에 따라 포인터를 이동하여 처리
        if (A[i] + A[j] < M) {
            i++;
        }
        else if (A[i] + A[j] > M) {
            j--;
        }
        else {
            count++;
            i++;
            j--;
        }
    }
    cout << count << "\n";
}

 

어렵게 느껴졌지만 생각보다 꽤 쉬운 문제였다. 이 문제에서 중요한 2가지 포인트가 있다.

 

1. 정렬

2. 포인터 위치

 

우선 정렬을 해주어야 대소관계를 명확히 할 수 있다.

그리고 포인터를 맨 앞에 하나와 맨 뒤에 하나씩 두며 간격을 좁혀가며 탐색을 한다.