졍's 기술 블로그

암기왕

티어 : Silver 4
시간 제한 : 2 초
메모리 제한 : 256 MB
알고리즘 분류 : 자료 구조, 정렬, 이분 탐색, 해시를 사용한 집합과 맵

문제

연종이는 엄청난 기억력을 가지고 있다. 그래서 하루 동안 본 정수들을 모두 기억 할 수 있다. 하지만 이를 믿을 수 없는 동규는 그의 기억력을 시험해 보기로 한다. 동규는 연종을 따라 다니며, 연종이 하루 동안 본 정수들을 모두 ‘수첩1’에 적어 놓았다. 그것을 바탕으로 그가 진짜 암기왕인지 알아보기 위해, 동규는 연종에게 M개의 질문을 던졌다. 질문의 내용은 “X라는 정수를 오늘 본 적이 있는가?” 이다. 연종은 막힘없이 모두 대답을 했고, 동규는 연종이 봤다고 주장하는 수 들을 ‘수첩2’에 적어 두었다. 집에 돌아온 동규는 답이 맞는지 확인하려 하지만, 연종을 따라다니느라 너무 힘들어서 여러분에게 도움을 요청했다. 동규를 도와주기 위해 ‘수첩2’에 적혀있는 순서대로, 각각의 수에 대하여, ‘수첩1’에 있으면 1을, 없으면 0을 출력하는 프로그램을 작성해보자.

입력

첫째 줄에 테스트케이스의 개수 T가 들어온다. 다음 줄에는 ‘수첩 1’에 적어 놓은 정수의 개수 N(1 ≤ N ≤ 1,000,000)이 입력으로 들어온다. 그 다음 줄에  ‘수첩 1’에 적혀 있는 정수들이 N개 들어온다. 그 다음 줄에는 ‘수첩 2’에 적어 놓은 정수의 개수 M(1 ≤ M ≤ 1,000,000) 이 주어지고, 다음 줄에 ‘수첩 2’에 적어 놓은 정수들이 입력으로 M개 들어온다. 모든 정수들의 범위는 int 로 한다.

출력

‘수첩2’에 적혀있는 M개의 숫자 순서대로, ‘수첩1’에 있으면 1을, 없으면 0을 출력한다.

예제 입출력

Algorithm

1. bisect 이용
2. 첫 번째 수첩 정렬
3. bisect_left와 bisect_right가 같으면 0, 다르면 1 출력

Code

import sys
from bisect import bisect_left, bisect_right

input = sys.stdin.readline

# 입력
T = int(input())

for _ in range(T):
    _ = input()
    array_n = list(map(int, input().split()))
    _ = input()
    array_m = list(map(int, input().split()))

    # array_n 정렬
    array_n.sort()
    for i in array_m:        
        # left와 right가 다르면 1, 같으면 0 출력
        if bisect_left(array_n, i) == bisect_right(array_n, i):
            print(0)
        else:
            print(1)

메모리: 203260 KB
시간: 3352 ms

나이순 정렬

티어 : Silver 5
시간 제한 : 3 초
메모리 제한 : 256 MB
알고리즘 분류 : 정렬

문제

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

입력

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

출력

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

예제 입출력

Algorithm

1. 나이와 이름을 튜플로 받아 리스트로 저장
2. lambda 함수를 이용해 나이순으로 정렬
    -> 나이가 같으면 이미 리스트에 저장되어 있던 순서대로 정렬되기 때문에 가입한 순으로 다시 정렬할 필요
        없음

Code

# 입력
N = int(input())
people = []
for _ in range(N):
    age, name = input().split()
    people.append((int(age), name))
    
# 나이순 정렬
people.sort(key = lambda x: x[0])
for i in people:
    print(i[0], i[1])

메모리: 45488 KB
시간: 4504 ms

좌표 정렬하기 2

티어 : Silver 5
시간 제한 : 1 초
메모리 제한 : 256 MB
알고리즘 분류 : 정렬

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

예제 입출력

Algorithm

1. 각 좌표를 Tuple로 받아 List에 저장
2. lambda 함수를 이용해 다중 조건 정렬 사용
    2.1. 1번 조건 : x[1]
    2.2. 2번 조건 : x[0]

Code

import sys
input = sys.stdin.readline

# 입력
N = int(input())
nums = []
for _ in range(N):
    nums.append(tuple(map(int, input().split())))
    
# 좌표 정렬
nums.sort(key = lambda x: (x[1], x[0]))
for i in nums:
    print(i[0], i[1])

메모리: 51636 KB
시간: 388 ms

좌표 정렬하기

티어 : Silver 5
시간 제한 : 1 초
메모리 제한 : 256 MB
알고리즘 분류 : 정렬

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

예제 입출력

Algorithm

1. 각 좌표를 튜플로 받아 리스트에 저장
2. lambda 함수를 이용해 다중 조건으로 sort
    2.1. 1번 조건 : x[0]
    2.2. 2번 조건 : x[1]

Code

import sys
input = sys.stdin.readline

# 입력
N = int(input())
nums = []
for _ in range(N):
    nums.append(tuple(map(int, input().split())))

# 좌표 정렬
nums.sort(key = lambda x:(x[0], x[1]))
for i in nums:
    print(i[0], i[1])

메모리: 51640 KB
시간: 404 ms

통계학

티어 : Silver 3
시간 제한 : 2 초
메모리 제한 : 256 MB
알고리즘 분류 : 수학, 구현, 정렬

문제

수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.

1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
4. 범위 : N개의 수들 중 최댓값과 최솟값의 차이

N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.

출력

첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.

둘째 줄에는 중앙값을 출력한다.

셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.

넷째 줄에는 범위를 출력한다.

예제 입출력

Algorithm

1. 각 숫자를 리스트에 저장   
    -> 음수값 나오는지 확인
2. 산술평균 = floor(sum(리스트) /N)
3. 중앙값 = 리스트 오름차순 정렬 후 list[N/2]
4. 최빈값 = 각 숫자들을 입력받을 때 Dictionary에 등장횟수와 함께 저장하고 value가 가장 큰 key를 출력
5. 범위 = 리스트[-1] - 리스트[0]

시간 초과
☞ 입력 시 readline()으로 입력받아 해결 

Code

import sys
input = sys.stdin.readline

# 입력
N = int(input())
nums = []
nums_dict = {}
for _ in range(N):
    num = int(input())
    nums.append(num)
    if num not in nums_dict:
        nums_dict[num] = 1
    else:
        nums_dict[num] += 1
        
# 산술평균
print(round(sum(nums)/N))

# 중앙값
nums.sort()
print(nums[N//2])

# 최빈값
max_ = max(nums_dict.values()) # 최대 등장 횟수 찾기
temp = []
for key, value in nums_dict.items():
    if value == max_: # value가 최대 등장 횟수와 같은 숫자 따로 빼두기
        temp.append(key)

# 최빈 값이 여러 개 인 경우
if len(temp) != 1:
    # 정렬 후 두 번째로 작은 값 출력
    temp.sort()
    print(temp[1])
else:
    print(temp[0])
    
# 범위
print(nums[-1] - nums[0])

메모리: 51480 KB
시간: 552 ms