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

문제

기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,

키보드
헤어드라이기
핸드폰 충전기
디지털 카메라 충전기
키보드
헤어드라이기
키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.

입력

첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다.

출력

하나씩 플러그를 빼는 최소의 횟수를 출력하시오.

예제 입출력

• 예제 입력 1

2 7
2 3 2 3 1 2 7

• 예제 출력 1

2

Solution

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

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int N, K;
    int answer = 0;
    vector<int> order;
	vector<int> plug;
    cin >> N >> K;
    for(int i = 0; i < K; i++){
        int input; cin >> input;
        order.push_back(input);
    }
    for(int i = 0; i < K; i++){
        if(plug.size() < N) {
            if(find(plug.begin(), plug.end(), order[i]) == plug.end()){
                plug.push_back(order[i]);    
        	}
        } else {
            if(find(plug.begin(), plug.end(), order[i]) == plug.end()){
                int index = -1;
                int last = -1;
                for(int j = 0; j < N; j++){
                    int tmp = 0;
                    for(int t = i + 1; t < K; t++){
                        if(order[t] == plug[j]) break;
                        tmp++;
                    }
                    if(last < tmp){
                        last = tmp;
                        index = j;
                    }
                }
                plug[index] = order[i];
                answer++;
            }
        }
    }
    cout << answer << '\n';
}

상당히 노가다성이 짙다. 하지만 입력 데이터 범위가 100 뿐이고 시간도 2초(?) 나 준다.

cin >> N >> K;
for(int i = 0; i < K; i++){
    int input; cin >> input;
    order.push_back(input);
}

우선 입력받은 사용 순서를 모두 벡터에 집어넣는다.

for(int i = 0; i < K; i++){
    if(plug.size() < N) {
        if(find(plug.begin(), plug.end(), order[i]) == plug.end()){
            plug.push_back(order[i]);
        }
    } else {
        if(find(plug.begin(), plug.end(), order[i]) == plug.end()){
            int index = -1;
            int last = -1;
            for(int j = 0; j < N; j++){
                int tmp = 0;
                for(int t = i + 1; t < K; t++){
                    if(order[t] == plug[j]) break;
                    tmp++;
                }
                if(last < tmp){
                    last = tmp;
                    index = j;
                }
            }
            plug[index] = order[i];
            answer++;
        }
    }
}

고려해야 할 상황은 크게 두가지다.
1. 플러그에 자리가 있는 경우
2. 플러그에 자리가 없는 경우

첫 번째 상황에서는 현재 꽂으려는 녀석이 플러그에 자리잡고 있지 않다면 꽂아도 좋다. 아니면 당연히 넘겨야 하는 상황이다.

두 번째 상황을 다시 둘로 나누면
1. 다음 차례에 꽂을 녀석이 지금 플러그에 있는 경우
2. 없는 경우

마지막 2번만 신경쓰면 쉽게 풀리는 문제다.

 if(find(plug.begin(), plug.end(), order[i]) == plug.end()){
    int index = -1;
    int last = -1;
    for(int j = 0; j < N; j++){
        int tmp = 0;
        for(int t = i + 1; t < K; t++){
            if(order[t] == plug[j]) break;
            tmp++;
        }
        if(last < tmp){
            last = tmp;
            index = j;
        }
    }
    plug[index] = order[i];
    answer++;
}

두 번째 상황의 두 번째 상황에서 이 로직이 수행된다.
현재 기점으로 플러그에 꽂혀있는 녀석들 중에 가장 마지막에 실행되는 녀석의 자리를 비우고 현재 순번을 꽂으면 그만이다.

생각보다 쉽게 풀렸으나 바로 떠오르지는 않았던 문제. 설마 이렇게 풀린다고? 싶었다만 2초라는 시간이 주어졌고 범위가 100까지니까 충분히 풀렸던 문제.