[BOJ알고리즘, C++]#11399_ATM, 정렬, 병합 정렬, 퀵 정렬

 

#1. 문제

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

11399번: ATM

 

---

 

#2. 풀이

 

1. 병합 정렬

[알고리즘]#3_정렬 알고리즘

 

[정의] : 병합 정렬은 분할-정복에 기반한 정렬 알고리즘으로, 주어진 배열을 더 이상 나눌 수 없는 작은 크기까지 분할하고, 병합 과정에서 정렬 작업을 재귀적으로 수행하여 최종적으로 하나의 정렬된 배열을 얻습니다.

 

2. 퀵 정렬

[알고리즘]#3_정렬 알고리즘

 

[정의] : 퀵 정렬 알고리즘은 분할-정복에 기반한 알고리즘으로, 주어진 배열을 피벗 원소를 기준으로 왼쪽은 피벗 원소보다 작은 원소들, 오른쪽은 큰 원소들로 정렬 및 분할 작업을 더 이상 나눌 수 없는 Base Case까지 재귀적으로 수행하여, 최종적으로 하나의 정렬된 배열을 얻습니다. 

 

3. 오름차순 정렬, 누적 합 계산!

 

1. 먼저, 돈을 인출하는데 필요한 시간을 기준으로 오름차순 정렬해줍니다.
2. 그리고, 해당 배열을 차례대로 순회하며 누적 합을 계산해 줍니다!

 

---

 

#1. 코드(병합 정렬)

 

// #1. 병합 정렬

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

int N;
vector<int> v;

void Merge(vector<int>& arr, int left, int mid, int right)
{
    int leftSize = mid - left + 1;
    int rightSize = right - mid;
    
    vector<int> leftHalf(leftSize);
    vector<int> rightHalf(rightSize);
    
    for(int i=0; i<leftSize; ++i)
        leftHalf[i] = arr[left+i];
    
    for(int i=0; i<rightSize; ++i)
        rightHalf[i] = arr[mid+1+i];
    
    int leftIdx = 0;
    int rightIdx = 0;
    int resIdx = left;
    
    while(leftIdx < leftSize && rightIdx < rightSize)
    {
        if(leftHalf[leftIdx] <= rightHalf[rightIdx])
            arr[resIdx++] = leftHalf[leftIdx++];
        else
            arr[resIdx++] = rightHalf[rightIdx++];
    }
    
    while(leftIdx < leftSize)
    {
        arr[resIdx++] = leftHalf[leftIdx++];
    }
    
    while(rightIdx < rightSize)
    {
        arr[resIdx++] = rightHalf[rightIdx++];
    }
    
}

void MergeSort(vector<int>& arr, int left, int right)
{
    if(left < right)
    {
        int mid = left + (right-left)/2;
        
        MergeSort(arr, left, mid);
        MergeSort(arr, mid+1, right);
        
        Merge(arr, left, mid, right);
    }
}

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

    cin >> N;
    
    while(N--)
    {
        int tmp;
        cin >> tmp;
        
        v.push_back(tmp);
    }
    
    // 병합 정렬
    MergeSort(v, 0, v.size()-1);
    
    // 누적 합 계산 및 결과 출력
    long long res = 0;
    for(int i=0; i<v.size(); ++i)
    {
        if(i != 0)
            v[i] += v[i-1];
        res += v[i];
    }
    
    cout << res;
    
    return 0;
}

 

#2. 코드(퀵 정렬)

 

// #2. 퀵 정렬

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

int N;
vector<int> v;

int Partition(vector<int>& v, int l, int r)
{
    int piv = v[r];
    int i = l-1;
    
    for(int j=l; j<r; ++j)
    {
        if(v[j] <= piv)
        {
            i++;
            swap(v[j], v[i]);
        }
    }
    swap(v[i+1], v[r]);
    
    return i+1; 
}

void QuickSort(vector<int>& v, int l, int r)
{
    if(l < r)
    {
        int piv = Partition(v, l, r);
        
        QuickSort(v, l, piv-1);
        QuickSort(v, piv+1, r);
    }
}

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

    cin >> N;
    
    v.resize(N);
    
    for(int i=0; i<N; ++i)
        cin >> v[i];
    
    // #1. 퀵 소트 수행
    QuickSort(v, 0, N-1);
    
    // #2. 결과 출력
    int res = v[0];
    for(int i=1; i<N; ++i)
    {
        v[i] = v[i-1] + v[i];
        res += v[i];
    }
    
    cout << res;
    
    return 0;
}

 

---