흙금이네 블로그

아이디어 투 포인터로 두 수의 차이가 M 이상일 때를 찾는다. 풀이 값의 차이만 알면 되므로 셋으로 입력 값들의 중복을 제거한 후 정렬된 리스트 A로 변환하고 그 길이를 N으로 저장한다. 투 포인터로 두 수의 차이와 M을 비교하며 M을 초과한 경우에 결과값과 비교해 더 작은 값으로 갱신해 나간다. 차이가 M인 경우는 더 이상 탐색할 필요가 없으므로 결과값을 M으로 저장 후 반복문을 종료한다. import sys input = sys.stdin.readline N, M = map(int, input().split()) A = set() for _ in range(N): A.add(int(input())) A = sorted(A) N = len(A) s = e = 0 res = A[-1]-A[0] while..

아이디어 원생들의 키 차이를 정렬하여 차이가 큰 값들을 제외하고 나머지 합을 구한다. 차이가 큰 값들은 어디에 포함되더라도 비용에 포함되므로 경계를 나눠 분리해야 비용을 줄일 수 있다. 풀이 원생들의 키 차이를 리스트 D에 추가하여 정렬한 후, K-1개의 큰 값들을 제외한 나머지 값들의 합을 구해 출력한다. 리스트 D의 길이는 N-1이므로 뒤에서부터 K-1개의 값을 제외하려면 D[:N-K]를 사용하면 된다. N, K = map(int, input().split()) H = list(map(int, input().split())) D = [] for i in range(N-1): D.append(H[i+1]-H[i]) D.sort() print(sum(D[:N-K])) 리스트 컴프리헨션보다 for문에서 a..

아이디어 현재 공보다 크기가 작은 공들의 크기 합과 크기가 작으면서 색이 같은 공들의 크기 합을 구하여 풀 수 있다. 풀이 공을 저장하는 리스트 balls에 공의 크기, 공의 색, 인덱스를 순서대로 추가하고 공의 크기 오름차순으로 정렬한다. 차례로 현재 공보다 작은 공의 크기를 total에 누적하여 더하고, sum_color 리스트에 공의 색에 따라 누적하여 더한다. 작은 공들의 크기 합에서 작으면서 색이 같은 공들의 크기 합을 뺀 값을 인덱스에 맞게 결과값으로 저장한다. import sys input = sys.stdin.readline N = int(input()) balls = [] for i in range(N): a, b = map(int, input().split()) balls.append(..