백준 1461번 도서관 C++

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

1461번: 도서관

 

양수, 음수로 나눠서 인접한 부분끼리 묶어서 가고 마지막에 가는 곳은 다시 돌아와도 되지 않기 때문에 돌아오는 비용을 빼준다.

 

#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, m;
vector<int> v;

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

    cin >> n >> m;
    v = vector<int>(n);
    for (int i = 0; i < n; i++) cin >> v[i];
    sort(v.begin(), v.end());
    int mid = -1;
    for (int i = 0; i < n; i++) {
        if (v[i] > 0) {
            mid = i;
            break;
        }
    }
    if (mid == -1) mid = n;
    int ans = 0;
    for (int i = 0; i < mid; i += m) {
        ans += abs(v[i] * 2);
    }
    for (int i = n - 1; i >= mid; i -= m) {
        ans += v[i] * 2;
    }
    ans -= max(abs(v[0]), v[n - 1]);
    cout << ans;
}