본문으로 이동
코드패스 by Devin.KR

문제 5258 · 메가 V

창구 대기 행렬의 우대 쌍

인증 제출 통계 · 5분마다 갱신 · 정답 비율 = 정답 제출 / 전체 제출
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1000ms / 테스트 (언어별 보정은 아래 표)메모리: 측정하지 않음(브라우저 채점)0000.00%

문제

한 은행 창구에 N명의 고객이 도착한 순서대로 줄을 서 있다. i번째로 도착한 고객에게는 우선도 점수 a_i가 매겨져 있다. 창구 직원은 줄에서 서로 다른 두 고객 i, j (i < j, 즉 i가 먼저 도착했다)를 짝지어 보고, 두 점수의 차이가 기준값 K를 넘는지에 따라 쌍을 세 종류로 나눈다. 먼저 온 고객의 점수가 늦게 온 고객의 점수보다 K를 초과하여 크면, 곧 a_i - a_j > K이면 이 쌍은 앞선 우대 쌍이다. 늦게 온 고객의 점수가 먼저 온 고객의 점수보다 K를 초과하여 크면, 곧 a_j - a_i > K이면 이 쌍은 뒤선 우대 쌍이다. 두 경우 모두 아니면 비슷한 쌍이다. 차이가 정확히 K인 쌍은 우대 쌍이 아니라 비슷한 쌍이며, 점수가 같은 고객끼리의 쌍도 비슷한 쌍이다. 한 쌍은 위 세 종류 중 정확히 하나에 속한다. 줄 전체에서 가능한 모든 쌍 N(N-1)/2개에 대해 앞선 우대 쌍, 뒤선 우대 쌍, 비슷한 쌍의 개수를 각각 구하시오. 줄을 반으로 나누어 각각 처리한 뒤 두 반쪽에 걸친 쌍을 세는 재귀적 방법을 쓰면 효율적으로 풀 수 있다.

입력

첫 줄에 정수 N, K가 공백으로 구분되어 주어진다. (1 ≤ N ≤ 200,000, 0 ≤ K ≤ 10^9) 둘째 줄에 a_1, a_2, ..., a_N이 공백으로 구분되어 주어진다. (0 ≤ a_i ≤ 10^9)

출력

한 줄에 앞선 우대 쌍의 수, 뒤선 우대 쌍의 수, 비슷한 쌍의 수를 공백으로 구분하여 출력한다.

예제 입력 1

5 2
5 1 4 9 2

예제 출력 1

3 4 3

예제 입력 2

1 0
7

예제 출력 2

0 0 0

예제 입력 3

4 0
3 3 1 3

예제 출력 3

2 1 3
힌트 펼치기
  • 구간을 반으로 나눠 각각 정렬하면서 왼쪽 원소와 오른쪽 원소 사이의 쌍만 새로 센다.
  • 왼쪽을 오름차순으로 훑으면 오른쪽에서 조건을 만족하는 경계가 단조롭게 움직이므로 투 포인터로 센다. 비슷한 쌍은 전체 쌍 수에서 나머지 둘을 빼서 구한다.

관련 서재 장

관련 장

출처: Devin.KR 자체 제작

알고리즘 분류분할 정복재귀병합 정렬투 포인터

풀이 작성

연습 채점은 공개 테스트로 실행합니다. 인증 제출은 코드를 봉인한 뒤 새 입력을 실행하고 서버에서 출력을 대조하며 코드와 판정을 저장합니다.

Python 기본 시간 제한 = 문제 시간 제한 × 3 + 2초. 문제별 예외는 아래 표에 반영됩니다. 표준 라이브러리만 지원합니다.

Ctrl+Enter: 연습 채점 · Tab: 들여쓰기 · Esc 후 Tab: 다음 요소로 이동. 언어별 초안은 이 브라우저에 저장됩니다. JavaScript 입력: require('fs').readFileSync(0, 'utf8'), readline 또는 input().

로그인하면 인증 제출을 사용할 수 있습니다.

풀이를 제출하면 여기에서 결과를 확인할 수 있습니다.

    연습 기록은 이 브라우저에 최근 50개를 저장합니다. 인증 기록은 내 제출에서 확인하세요.

    이 브라우저의 연습 기록

      언어별 제한

      테스트 케이스마다 새 전역에서 실행하며 준비 시간은 제외합니다. 시간 초과된 출력은 검사하지 않습니다. 브라우저·기기 성능에 따라 실행 시간이 달라집니다.

      언어엔진(버전)시간 제한메모리
      javascript사용자 브라우저 JavaScript 엔진 (버전은 브라우저에 따라 다름)1000ms / 테스트측정하지 않음(브라우저 채점)
      pythonPyodide 314.0.7 (CPython 3.14 WASM)5000ms / 테스트측정하지 않음(브라우저 채점)