[HackerRank] Mini-Max Sum (C++, Python)

---

C++

#include <bits/stdc++.h>

using namespace std;

vector<string> split_string(string);

// Complete the miniMaxSum function below.
void miniMaxSum(vector<int> arr) {
    __int64_t max=0, min=0;
    sort(arr.begin(), arr.end());
    for(int i=0, j=arr.size()-1; i < arr.size()-1 && j > 0; i++, j--){
        max += arr[j]; min += arr[i];
    }
    cout << min << " " << max << endl;
}

int main()
{
    string arr_temp_temp;
    getline(cin, arr_temp_temp);

    vector<string> arr_temp = split_string(arr_temp_temp);

    vector<int> arr(5);

    for (int i = 0; i < 5; i++) {
        int arr_item = stoi(arr_temp[i]);

        arr[i] = arr_item;
    }

    miniMaxSum(arr);

    return 0;
}

vector<string> split_string(string input_string) {
    string::iterator new_end = unique(input_string.begin(), input_string.end(), [] (const char &x, const char &y) {
        return x == y and x == ' ';
    });

    input_string.erase(new_end, input_string.end());

    while (input_string[input_string.length() - 1] == ' ') {
        input_string.pop_back();
    }

    vector<string> splits;
    char delimiter = ' ';

    size_t i = 0;
    size_t pos = input_string.find(delimiter);

    while (pos != string::npos) {
        splits.push_back(input_string.substr(i, pos - i));

        i = pos + 1;
        pos = input_string.find(delimiter, i);
    }

    splits.push_back(input_string.substr(i, min(pos, input_string.length()) - i + 1));

    return splits;
}

• c++에서 integer의 범위는 다음과 같고, test case 10개는 output이 integer 범위를 넘는 경우들이라서 max와 min을 int64로 줘야한다.

• arr를 sort로 정렬한 후, min은 처음(0번째 원소)부터 3번째 원소까지, max는 마지막 원소(4번째 원소)부터 1번째 원소까지 각 min와 max에 더해가면서 구한다.
• 반복문의 경우 max와 min에서 각각 다른 인덱스에 접근해서 구하기 위해 두 개의 변수를 줬다.
• 이번 문제는 반환이 아닌 출력이라 cout으로 min과 max 사이에 space 하나를 주고 출력해줬다.

 

Python

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the miniMaxSum function below.
def miniMaxSum(arr):
    arr.sort()
    max=min=0
    for i, j in zip(range(5), range(4,0,-1)):
        max += arr[j]
        min += arr[i]
    print(str(min)+" "+str(max))
        

if __name__ == '__main__':
    arr = list(map(int, input().rstrip().split()))

    miniMaxSum(arr)

• C++에서와 똑같이 정렬한 후(sort 사용), min은 0~3번째를, max는 4~1을 더해서 구한다.
• python에서는 for문에서 변수 i,j의 범위를 zip으로 나눠서 줬다.