[Algorithm]Baekjoon 1461번 :: 도서관

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

 

#include <iostream>
#include <string>
#include <map>
#include <vector>
#include <algorithm>
#include <math.h>
#include <queue>
#include <functional>
#include <sstream>
#include <memory.h>
#include <deque>
#include <set>
#include <unordered_map>
#include <stack>
#include <numeric>
#include <climits>
#include <bitset>
#include <cmath>
 
using namespace std;
 
int n, m, a;
vector<int> positiveNums;
vector<int> negativeNums;
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
 
    cin >> n >> m;
 
    for (int i = 0; i < n; ++i)
    {
        cin >> a;
 
        if (a > 0)
        {
            positiveNums.push_back(a);
        }
        else
        {
            negativeNums.push_back(a);
        }
    }
 
    sort(positiveNums.rbegin(), positiveNums.rend());
    sort(negativeNums.begin(), negativeNums.end());
 
    int answer = 0;
    int positiveIndex = 0;
    int negativeIndex = 0;
 
    if (negativeNums.size() == 0)
    {
        answer = positiveNums[0];
        positiveIndex = m;
    }
    else if (positiveNums.size() == 0)
    {
        answer = negativeNums[0] * -1;
        negativeIndex = m;
    }
    else
    {
        if (negativeNums[0] * -1 >= positiveNums[0])
        {
            answer = negativeNums[0] * -1;
            negativeIndex = m;
        }
        else if (positiveNums.size() > 0)
        {
            answer = positiveNums[0];
            positiveIndex = m;
        }
    }
 
    while (negativeIndex < negativeNums.size())
    {
        answer += negativeNums[negativeIndex] * -1 * 2;
 
        negativeIndex += m;
    }
 
    while (positiveIndex < positiveNums.size())
    {
        answer += positiveNums[positiveIndex] * 2;
 
        positiveIndex += m;
    }
 
    cout << answer;
}
 
 

 

이 문제에서의 중요포인트는 마지막책을 갖다 놓으면 다시 원점위치로 되돌가가지 않아도 되는 것.

그러니 제일 멀리있는 위치를 기준으로 한 m개로 구성된 그룹은 제일 마지막에 가져다 놓으면 된다.