[Programmers] 요격 시스템

요격 시스템

A 나라와 B 나라가 싸우고 있는 이 세계는 2 차원 공간으로 이루어져 있습니다. A 나라가 발사한 폭격 미사일은 x 축에 평행한 직선 형태의 모양이며 개구간을 나타내는 정수 쌍 (s, e) 형태로 표현됩니다. B 나라는 특정 x 좌표에서 y 축에 수평이 되도록 미사일을 발사하며, 발사된 미사일은 해당 x 좌표에 걸쳐있는 모든 폭격 미사일을 관통하여 한 번에 요격할 수 있습니다. 단, 개구간 (s, e)로 표현되는 폭격 미사일은 s와 e에서 발사하는 요격 미사일로는 요격할 수 없습니다. 요격 미사일은 실수인 x 좌표에서도 발사할 수 있습니다.
각 폭격 미사일의 x 좌표 범위 목록 targets이 매개변수로 주어질 때, 모든 폭격 미사일을 요격하기 위해 필요한 요격 미사일 수의 최솟값을 return 하도록 solution 함수를 완성해 주세요.

풀이

오름차순 정렬을 해야겠다는 것까지는 알았는데 그 뒤를 구현하지 못했다. 찾아본 결과 2개의 방법이 있다.

• (s,e) 둘 다 오름차순 정렬하는 방법
• e만 오름차순 정렬하는 방법

코드

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<vector<int>> targets) {
    int n = targets.size();
    // s 기준 오름차순 정렬, s 같으면 e 기준 오름차순 정렬
    sort(targets.begin(), targets.end(), 
         [](vector<int> &a, vector<int> &b){
            if(a[0]==b[0]) return a[1]<b[1];
            return a[0]<b[0];
    });
    
    int answer = 1;
    int end = targets[0][1];
    for(int i=1; i<n; i++){
        if(targets[i][0] >= end){ // end보다 현재 시작이 크거나 같으면 새로운 미사일 필요
            answer++;
            end = targets[i][1];
            continue;
        }
        end = min(end, targets[i][1]); // end보다 현재 끝이 작으면 갱신
    }
    
    return answer;
}

#include <bits/stdc++.h>

using namespace std;

int solution(vector<vector<int>> targets) {
    sort(targets.begin(), targets.end(), 
          [](vector<int>& a, vector<int>& b) {
              return a[1] < b[1];
    });

    int ans = 0;
    int end = -1; 
    for (auto& tar: targets) { 
      if (tar[0] >= end) { // end보다 현재 시작이 크거나 같으면 새로운 미사일 필요
        end = tar[1];
        ans++;
      }
    }

  	return ans;
}

정리

(s, e)에서 e를 기준으로 풀어나가야 했다.

참고