143.両配列の和
13076 ワード
白駿
1. Python
from collections import defaultdict
input = sys.stdin.readline
n = int(input())
a = int(input())
alist = list(map(int, input().split()))
b = int(input())
blist = list(map(int, input().split()))
adic = defaultdict(int)
answer = 0
for i in range(a):
for j in range(i, a):
adic[sum(alist[i:j + 1])] += 1
for i in range(b):
for j in range(i, b):
answer += adic[n - sum(blist[i:j + 1])]
print(answer)
分割コードの使用
from sys import stdin
from bisect import *
def solve():
t = int(stdin.readline().strip())
n = int(stdin.readline().strip())
A = [0] + list(map(int, stdin.readline().split()))
m = int(stdin.readline().strip())
B = [0] + list(map(int, stdin.readline().split()))
for i in range(1, n + 1):
A[i] += A[i - 1]
for i in range(1, m + 1):
B[i] += B[i - 1]
temp = []
for i in range(m + 1):
for j in range(i + 1, m + 1):
temp.append(B[j] - B[i])
temp.sort()
ans = 0
for a1 in range(n + 1):
for a2 in range(a1 + 1, n + 1):
now = A[a2] - A[a1]
target = t - now
foo = bisect_left(temp, target)
bar = bisect_right(temp, target)
ans += bar - foo
return ans
print(solve())
ソースReference
この問題について(143.両配列の和), 我々は、より多くの情報をここで見つけました https://velog.io/@corone_hi/143.-두-배열의-합テキストは自由に共有またはコピーできます。ただし、このドキュメントのURLは参考URLとして残しておいてください。
Collection and Share based on the CC Protocol