백준 18513번 샘터 C++

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

18513번: 샘터

 

샘터 주변의 위치부터 탐색하면 된다.

 

샘터를 기준으로 -1, +1씩 검사하면서 지을 수 있으면 짓고 큐에 그 위치를 넣고 하면서 k개의 집을 지었을 때 불행도를 합치면 된다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, k;
int64 ans = 0;

unordered_set<int> us;

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

    cin >> n >> k;
    queue<pair<int,int>> q;
    for (int i = 0; i < n; i++) {
        int loc;
        cin >> loc;
        q.push({loc,0});
        us.insert(loc);
    }

    int cnt = 0;
    while (!q.empty()) {
        auto curr = q.front();
        q.pop();

        int back = curr.first - 1;
        auto it = us.find(back);
        if (it == us.end()) {
            ans += abs(curr.second - 1);
            cnt++;
            if (cnt == k) break;
            q.push({ back,curr.second - 1 });
            us.insert(back);
        }

        int front = curr.first + 1;
        it = us.find(front);
        if (it == us.end()) {
            ans += abs(curr.second + 1);
            cnt++;
            if (cnt == k) break;
            q.push({ front,curr.second + 1 });
            us.insert(front);
        }
    }

    cout << ans;
}