Safe Range Sum Queries
Python
Medium
2 views
Problem Description
Read n numbers and q queries l r (0-based inclusive). For each query output sum. If query is out of range or invalid, output ERROR for that query.
Input Format
First line n q. Second line n ints. Next q lines l r.
Output Format
q lines sums or ERROR.
Sample Test Case
Input:
5 3
1 2 3 4 5
0 2
2 10
x 1
Official Solution
import sys
lines=sys.stdin.read().splitlines()
if len(lines)<2: sys.exit(0)
first=lines[0].split()
if len(first)<2: sys.exit(0)
n=int(first[0]); q=int(first[1])
a=list(map(int,lines[1].split()[:n]))
pref=[0]*(n+1)
for i in range(n):
pref[i+1]=pref[i]+a[i]
out=[]
for i in range(q):
try:
l,r=map(int,(lines[2+i] if 2+i<len(lines) else '').split())
if l<0 or r<0 or l>r or r>=n:
out.append('ERROR')
else:
out.append(str(pref[r+1]-pref[l]))
except Exception:
out.append('ERROR')
sys.stdout.write('\
'.join(out))
Solutions (0)
No solutions submitted yet. Be the first!
No comments yet. Start the discussion!