ps:problems:boj:16139
목차
인간-컴퓨터 상호작용
| ps | |
|---|---|
| 링크 | acmicpc.net/… |
| 출처 | BOJ |
| 문제 번호 | 16139 |
| 문제명 | 인간-컴퓨터 상호작용 |
| 레벨 | 실버 1 |
| 분류 |
누적합 |
| 시간복잡도 | O(n + q) |
| 인풋사이즈 | n<=200,000, q<=200,000 |
| 사용한 언어 | Python |
| 제출기록 | 79192KB / 708ms |
| 최고기록 | 580ms |
| 해결날짜 | 2022/05/31 |
| 태그 | |
풀이
- 부분합 쿼리 문제. 알파벳별로 누적합을 각각 만들어주고 쿼리에서 물어보는 알파벳의 누적합에서 계산해주면 된다.
- 누적합을 최대 26개 만들어야 하니까 O(|S|*26)이 걸리고, 쿼리는 각 쿼리당 O(1) 에 처리가능하다. 알파벳의 갯수 26을 상수로 보면, O(|S| + q).
코드
"""Solution code for "BOJ 16139. 인간-컴퓨터 상호작용".
- Problem link: https://www.acmicpc.net/problem/16139
- Solution link: http://www.teferi.net/ps/problems/boj/16139
Tags: [Prefix sum]
"""
import sys
def main():
S = sys.stdin.readline().rstrip()
prefix_sums = {}
for c in set(S):
prefix_sums[c] = ([ps := 0] +
[ps := (ps + 1 if c == s_i else ps) for s_i in S])
q = int(sys.stdin.readline())
for _ in range(q):
alpha, l, r = sys.stdin.readline().split()
try:
psum = prefix_sums[alpha]
print(psum[int(r) + 1] - psum[int(l)])
except KeyError:
print('0')
if __name__ == '__main__':
main()
ps/problems/boj/16139.txt · 마지막으로 수정됨: 2022/11/18 01:56 저자 teferi

토론