四大文化赛道完整展开
02-solution/src/python/main.py
main.py
站内文件视图直接读取仓库内容,Markdown 使用文档排版渲染,其余文本文件保持原始排版,方便校对训练证据链。
文件类型.py
10-cases/s4-jh-06-dye-batch-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)
for index in range(1, n + 1):
prefix[index] = prefix[index - 1] + values[index]
lines = []
best_query = 1
best_sum = -1
for query_id in range(1, q + 1):
left = next(it)
right = next(it)
total = prefix[right] - prefix[left - 1]
lines.append(f"query_{query_id}={total}")
if total > best_sum:
best_sum = total
best_query = query_id
lines.append(f"max_query={best_query}")
return "\n".join(lines)
if __name__ == "__main__":
sys.stdout.write(solve(sys.stdin.read()).strip())
sys.stdout.write("\n")