[실버3] 2108. 통계학 (Java)

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

2108번: 통계학

SSAFY10기 여러분 IM, A형 테스트 고생 많으셨습니다~!

---

※  문제를 요약하면 아래와 같습니다.

▶  N개의 숫자가 주어짐 ( 단, N은 홀수)

▶ 기본 통계값들 중 아래 4가지 출력

    : 산술평균 - N개의 수들의 합을 N으로 나눈 값

    : 중앙값 - N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값

    : 최빈값 - N개의 수들 중 가장 많이 나타나는 값

    : 범위 : N개의 수들 중 최댓값,최소값의 차이

▶ Input : 첫줄에 수의 개수 N이 주어지며 다음 N개의 줄에는 정수들이 주어진다.

▶ Output : 첫째 줄에 산술평균 출력 ( 소수점 이하 첫째 자리 반올림 )

                 : 둘째 줄에 중앙값 출력

                 : 셋째 줄에는 최빈값 출력 ( 여러개 있을 땐 두번째로 작은 값)

                 : 넷째 줄에는 범위 출력

---

◈ Input - 1

5
1
3
8
-2
2

◈ Output - 1

2
2
1
10

◈ Input - 2

5
-1
-2
-3
-1
-2

◈ Output - 2

-2
-2
-1
2

---

 ◎  코드 작성 전, 아래와 같이 솔루션을 정리하였습니다.

▶ 산술평균 - 주어진 모든 수의 합 / N 에 Math.round()사용 (소수점 첫째자리 반올림)

▶ 중앙값 - 홀수개 이므로 정렬 후 N / 2 index 값 출력

▶ 최빈값 - N개의 수들을 인덱스로 받아 카운트, 최빈값이 여러 개일 경우 2번째로 작은 수 출력

▶ 범위값 - 배열 정렬 후 가장 첫번째값과 가장 끝값의 차를 구하여 출력

▶ 음수의 경우, 최빈값 배열을 따로 만들어 인덱스 관리

---

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader bReader = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bWriter = new BufferedWriter(new OutputStreamWriter(System.out));

		int N = Integer.parseInt(bReader.readLine());
		int[] narrCount = new int[500000]; // 양수 카운트
		int[] narrNegativeCount = new int[500000]; // 음수 카운트

		int[] narrTemp = new int[N];
		int nSum = 0;
		for (int i = 0; i < N; i++) {
			narrTemp[i] = Integer.parseInt(bReader.readLine());
			if (narrTemp[i] > 0)
				narrCount[narrTemp[i]]++;// 빈도수 구하기 위해 해당 수 카운트
			else
				narrNegativeCount[narrTemp[i] * -1]++;
			nSum += narrTemp[i]; // 평균값 구하기 위한 sum
		}

		Arrays.sort(narrTemp);

		int nAverage = (int) Math.round((double) nSum / (double) N);
		int nMiddle = narrTemp[N / 2];
		int nMax = 0;
		int nIndex = 0;

		// 맥스 빈도수 찾기
		for (int i = 0; i < N; i++) {
			if (narrTemp[i] > 0) {
				if (nMax < narrCount[narrTemp[i]]) {
					nMax = narrCount[narrTemp[i]];
					nIndex = narrTemp[i];
				}
			} else {
				if (nMax < narrNegativeCount[narrTemp[i] * -1]) {
					nMax = narrNegativeCount[narrTemp[i] * -1];
					nIndex = narrTemp[i];
				}
			}
		}

		boolean bSame = false;
		
		// 맥스 빈도수 같을 경우 2번째로 작은 값 출력
		for (int i = 0; i < N; i++) {
			if(i != N-1) {
				if(narrTemp[i] == narrTemp[i+1])
					continue;
			}
			
			if (narrTemp[i] > 0) {
				if (nMax == narrCount[narrTemp[i]]) {
					if (!bSame)
						bSame = true;
					else {
						nIndex = narrTemp[i];
						break;
					}
				}
			} else {
				if (nMax == narrNegativeCount[narrTemp[i] * -1]) {
					
					if (!bSame)
						bSame = true;
					else {
						nIndex = narrTemp[i];
						break;
					}
				}
			}
		}
		
		int nRange = narrTemp[N - 1] - narrTemp[0];

		bWriter.write(nAverage + "\n");
		bWriter.write(nMiddle + "\n");
		bWriter.write(nIndex + "\n");
		bWriter.write(nRange + "\n");

		// bWriter.write(String.valueOf(narrTemp));
		bWriter.flush();
		bWriter.close();
	}
}

 으 최빈값 때문에 if문 덕지덕지라 보기가 싫네요..  근데 너무 수정하기에는 너무 피곤..

우리 반 IM테스트를 전부 통과한 것 같아서 기분이 너무 좋습니다~~~ 4FY 화이팅~

읽어주셔서 감사합니다!

Good Luck! (피드백 감사합니다!)