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

문제 5274 · 메가 IV

자전거 대여소 계층 비교

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

문제

공유 자전거 서비스의 대여소 N곳은 1번 중앙 대여소를 뿌리로 하는 트리 모양으로 관리된다. 2번부터 N번까지의 대여소 i 는 상위 대여소 p_i 를 하나씩 가지며 p_i < i 이다. 각 대여소 i 에는 거치대가 a_i 개 있다. 어떤 대여소 v 가 대여소 u 의 상위 대여소란, v 가 u 의 부모이거나 부모의 부모처럼 위로 거슬러 올라가면 닿는 대여소를 말한다. u 자신은 포함하지 않는다. 운영팀은 거치대 수가 상위 대여소보다 더 많은 '확장형' 관계를 점검하려 한다. 상위 대여소 v 와 그 아래 대여소 u 의 쌍 (v, u) 중 a_v < a_u 를 만족하는 쌍을 '확장 쌍'이라 한다. 거치대 수가 같은 쌍은 확장 쌍이 아니다. 대여소 u 마다 u 의 상위 대여소 중 a_v < a_u 인 것의 수를 c_u 라 하자. 모든 확장 쌍의 수를 구하고, c_u 가 최대인 대여소와 그 최댓값도 구하는 프로그램을 작성하라. 최댓값이 같은 대여소가 여럿이면 번호가 가장 작은 대여소를 고른다. 모든 c_u 가 0 이면 1번 대여소와 0 을 출력한다. 트리가 한 줄처럼 깊어질 수 있으므로 재귀 호출이 너무 깊어지지 않게 주의해야 한다.

입력

첫 줄에 N 이 주어진다. (1 ≤ N ≤ 200,000) 둘째 줄에 p_2 … p_N 이 공백으로 주어진다. (1 ≤ p_i < i) N = 1 이면 이 줄은 비어 있다. 셋째 줄에 a_1 … a_N 이 공백으로 주어진다. (1 ≤ a_i ≤ 10^9, 정수)

출력

한 줄에 확장 쌍의 수, c_u 가 최대인 대여소 번호, 그 최댓값을 공백으로 구분해 출력한다.

예제 입력 1

5
1 1 2 2
3 5 2 6 1

예제 출력 1

3 4 2

예제 입력 2

1

7

예제 출력 2

0 1 0

예제 입력 3

4
1 2 3
1 2 2 3

예제 출력 3

5 4 3
힌트 펼치기
  • 루트에서 내려가며 현재 경로의 값만 펜윅 트리에 담아 두고, 올라올 때 되돌린다.
  • 값은 좌표 압축하고, 엄격히 작은 값의 개수를 질의한다.

관련 서재 장

관련 장

출처: Devin.KR 자체 제작

알고리즘 분류DFS트리펜윅 트리백트래킹

풀이 작성

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

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 / 테스트측정하지 않음(브라우저 채점)