6-g0rnn

[g0rnn]

🔗 문제 링크

통계학

✔️ 소요된 시간

1h

✨ 수도 코드

저번부터 구현이 약한 것 같아. 구현을 연습해보고 있습니다. 간단해서 10분만에 풀고 다른 문제 올리려고 했는데 1시간이 넘어가게 되어 그냥 올리겠습니다.

파이썬에서는 얼마나 어려울지 모르겠지만.. cpp에서는 귀찮은게 몇가지 있었습니다.

저는 최빈값을 구하는거에서 어려웠습니다. 누구는 그냥 8000크기의 배열을 만드는데, 저는 그런 스타일이 아니라서 함수로 구현했는데 그거때문에 시간을 좀 잡아먹었습니다...ㅜ

나머지는 쉬우니 최빈값을 함수로 만들어 구현하는 수도코드만 소개할게요

0. 정렬되어 있는 배열을 사용한다.
1. i번째와 i+1번째가 같은지 확인한다.
2. 같으면 빈도수를 올린다.
3. 다르면 찾은 최빈값을 업데이트하거나 넘어간다.
  3-1. 이전 빈도수 보다 더 많은 빈도수를 찾으면 최빈값을 업데이트한다.
    -> 항상 제일 작은 최빈값을 유지한다.
  3-2. 처음으로 중복되는 빈도값을 찾으면 두번째로 작은 최빈값을 업데이트한다. 

📚 새롭게 알게된 내용

cpp에서는 부동 소수점을 사용하기 때문에 -0이 존재할 수 있다는 것을 알게되었습니다. 따라서 round를 사용할 때 주의해야합니다.

[kangrae-jo]

저도 잠자기 전에 빨리 리뷰하고 자야겠다는 생각이었는데요....
최빈값이 가지말라고 하네요...

저도 추가 배열을 안쓰고 싶었지만 결국 썼습니다.

살짝 머리를 굴려서 {count, index} 로 저장을 했습니다.
이렇게 하면 비교는 count로, 나중에 return값은 index로 해서 답을 쉽게 찾을 수 있어요.

그리고 살짝 어려움이 있었던건 compareTo를 정의하는 부분인데요.
로직은 머리에 있었지만 평소에 잘 써보지 않아서 지피티 찬스 썼습니다.
다음에는 검색 없이 써먹을 수 있을 것 같네요.

또 음수에서 반올림하는 추가적인 로직이 필요할 줄 알았는데 그냥 round 하면 되네요.

고생하셨습니다.

CPP CODE

#include <algorithm>
#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

int N;

bool compareTo(pair<int, int> a, pair<int, int> b) {
    if (a.first == b.first) return a.second < b.second;
    return a.first > b.first;
}

int solution1(vector<int> s) {
    int sum = 0;
    for (int i : s) sum += i;

    float average = (float) sum / N;
    return round(average); 
}

int solution2(vector<int> s) {
    return s[N / 2];
}

int solution3(vector<int> s) {
    vector<pair<int, int>> temp(8001, {0, 0});
    for (int i : s) {
        temp[i + 4000].first++;
        temp[i + 4000].second = i;
    }

    sort(temp.begin(), temp.end(), compareTo);

    if (temp[0].first == temp[1].first) return temp[1].second;
    return temp[0].second;
}

int solution4(vector<int> s) {
    return s[N - 1] - s[0];
}

int main() {
    cin >> N;
    vector<int> s(N);
    for (int i = 0; i < N; i++) cin >> s[i];
    
    sort(s.begin(), s.end());
    cout << solution1(s) << endl;
    cout << solution2(s) << endl;
    cout << solution3(s) << endl;
    cout << solution4(s) << endl;

    return 0;
}

[kangrae-jo]

처음에 tmp를 arr[0]로 초기화 하면, for문을 1부터 돌릴 수 있을 것 같아요.
따라서 arr[i] == arr[i-1]로 비교할 수 있고, else인 상황에서 tmp+=1 을 안해도 됐을 것 같아요.
(정확한건 뒤의 로직도 이해해야 하지만.. 조금 패스하겠습니다 ㅎㅎ)
if와 else모두 tmp를 증가시키니까 살~짝 더 복잡해진듯 합니다.

구현 완전 빡세네요...
저도 배열 만드는건 제 타입이 아니어서 균호님처럼 시도해봤는데,
자꾸 고려해야할 사항들이 생기길래 포기했습니다.

[g0rnn]

float과 double 모두 부동소수점으르 사용하여 -0이 존재하고 콘솔에 0이 아니라 -0으로 출력됩니다.
하지만 solution1의 반환 값이 int형이어서 자동 형변환으로 인해 0으로 변환되어 출력된것 같습니다. :)

[kangrae-jo]

엇 그렇군요. -0이 있는 걸 모르고 있었네요.
그리고 함수 반환 타입도 무시하고 있었네요. ..감사합니다~

[wnsmir]

파이썬에서는 collections.Counter를 활용해 최빈값을 구하는 것이 매우 편리했습니다. C++에서 8000 크기의 배열을 직접 생성하는 것처럼 고정된 배열을 사용하는 대신, 파이썬은 데이터의 범위를 몰라도 동적으로 처리할 수 있는 장점이 있습니다. 저는 이 점을 활용해 효율적으로 최빈값을 계산했습니다.

from collections import Counter
import sys

input = sys.stdin.read
data = input().split()
N = int(data[0])
numbers = list(map(int, data[1:]))

# 1. 산술평균
mean = round(sum(numbers) / N)

# 2. 중앙값
numbers.sort()
median = numbers[N // 2]

# 3. 최빈값
count = Counter(numbers)
most_common = count.most_common()
most_common.sort(key=lambda x: (-x[1], x[0]))  # 빈도순 정렬 후 값 기준 정렬

if len(most_common) > 1 and most_common[0][1] == most_common[1][1]:
    mode = most_common[1][0]  # 두 번째로 작은 최빈값
else:
    mode = most_common[0][0]  # 첫 번째 최빈값

# 4. 범위
range_value = numbers[-1] - numbers[0]

# 결과 출력
print(mean)
print(median)
print(mode)
print(range_value)

1. Counter의 유용성
   최빈값 계산에서 collections.Counter가 매우 유용하다고 느꼈습니다. C++처럼 직접 배열을 생성하거나 반복문을 이용해 빈도를 세는 과정 없이 단순히 Counter를 사용해 빠르게 빈도를 계산할 수 있었습니다.

2. 부동 소수점 처리
   파이썬에서는 round() 함수가 음수를 처리할 때도 올바르게 반올림되므로 따로 부호를 확인하거나 추가적인 예외 처리를 하지 않아도 됐습니다.