四大文化赛道完整展开
02-solution/src/python/main.py
main.py
站内文件视图直接读取仓库内容,Markdown 使用文档排版渲染,其余文本文件保持原始排版,方便校对训练证据链。
文件类型.py
10-cases/s1-jh-05-heritage-traffic-query/02-solution/src/python/main.py
import sys
def solve(data: str) -> str:
tokens = list(map(int, data.split()))
if not tokens:
return ""
it = iter(tokens)
n = next(it)
q = next(it)
values = [0] + [next(it) for _ in range(n)]
prefix = [0] * (n + 1)
best_slot = 1
for index in range(1, n + 1):
prefix[index] = prefix[index - 1] + values[index]
if values[index] > values[best_slot]:
best_slot = index
lines = []
for query_id in range(1, q + 1):
left = next(it)
right = next(it)
lines.append(f"query_{query_id}={prefix[right] - prefix[left - 1]}")
lines.append(f"peak_slot={best_slot} {values[best_slot]}")
return "\n".join(lines)
if __name__ == "__main__":
sys.stdout.write(solve(sys.stdin.read()).strip())
sys.stdout.write("\n")