백준2108번 파이썬 #12정렬-4)통계학

 

내 풀이 1(오답)

from collections import Counter as cc

    n=int(input())
    num=[int(input()) for x in range(n)]
    print(sum(num)//n)
    print(sorted(num)[n//2])
    cnt=cc(num)
    center=cnt.most_common()
    if len(center)>1:
        if center[0][1]==center[1][1]:
            print(center[1][0])
        else:
            print(center[0][0])
    else:
        print(center[0][0])

    print(abs(max(num)-min(num)))

오류 :

발생 원인:

most_common() 한 center 리스트의 정렬순서

1. 빈도 기준

2. 값 기준 

이렇게 되어야 하는데, 값 기준 정렬이 되지 않는다.

ex) 

입력	5  -1 -2 -3 -1 -2
center 정렬	[(-1, 2), (-2, 2), (-3, 1)]

 

 

내풀이 2(오답)

from collections import Counter as cc
if __name__=='__main__':
    n=int(input())
    num=[int(input()) for x in range(n)]
    print(sum(num)//n)
    print(sorted(num)[n//2])
    cnt=cc(num)
    center=cnt.most_common()
    if n==1:
        print(center[0])
    center=sorted(center,key=lambda x:(x[1],-x[0]),reverse=True)
    if len(center)>1:
        if center[0][1]==center[1][1]:
            print(center[1][0])
        else:
            print(center[0][0])
    else:
        print(center[0][0])

    print(abs(max(num)-min(num)))

 

이렇게 고쳣는데 시간초과

 

발생원인:

center sort를 또 정렬하고 할 필요가 없음

시간 초과를 고치기 위해 sys를 써야함

 

 

 

 

내 풀이 (최종)

from collections import Counter as cc
import sys
if __name__=='__main__':
    n=int(input())
    num=sorted([int(sys.stdin.readline()) for x in range(n)])
    print(round(sum(num)/n))
    print(num[n//2])
    cnt=cc(num)
    center=cnt.most_common(2)		#최빈값 2개만 center로 뽑기
    if n==1:
        print(center[0][0])			#n이 1일때 (총1개)
    else:
        if center[0][1]==center[1][1]:	#빈도수 같은게 있을땐 작은 값
            print(center[1][0])
        else:
            print(center[0][0])
    print(num[n-1]-num[0] if n!=1 else 0)

- most_common(2)  : 2번째 작은수 까지만 필요하니 최빈 2개만 뽑기

- num리스트를 입력받으면서 바로정렬 : most_common할때 따로 정렬할 필요가 없음

 

 

 

배운 것

 

※ collections 의 Counter 라이브러리 

from collections import Counter as cc

Counter를 cc로 해서 쓰겠다.

as cc 생략 가능

object = Counter( 입력배열 ) 출력배열 =object.most_common()

object= cc( 입력배열) 출력배열=object.most_common()

most_common(2) : 최빈값 2개만 출력

 

출력 순서 기준

빈도 큰 순서대로( 빈도 내림차순)

ex)

[(-1, 2), (-2, 2), (-3, 1)]

 

※ lambda

center=sorted(center,key=lambda x:(x[1],-x[0]),reverse=True)

[(-1, 2), (-2, 2), (-3, 1)]

[(-2, 2), (-1, 2), (-3, 1)]

1)  x[1]기준 내림차순 정렬

2)  x[0] 기준 오름차순 정렬

 

실제로 쓰진 않았지만, 정렬과 lambda를 공부할 수 있었음