====== 나누기 ====== ===== 풀이 ===== * 우선 당연하게도, 수열의 합이 4의 배수가 아니라면 4조각으로 나누는 것이 불가능하니 그것부터 체크하자. * 합을 4로 나눈 값이 q라고 하자. A의 누적합 배열을 P라고 한다면(P[i] = A[1]+..+A[i]), 이 문제는 P[i]=q, P[j]=2*q, P[k]=3*q 이고 1≤i """Solution code for "BOJ 21757. 나누기". - Problem link: https://www.acmicpc.net/problem/21757 - Solution link: http://www.teferi.net/ps/problems/boj/21757 """ def main(): N = int(input()) # pylint: disable=unused-variable A = [int(x) for x in input().split()] q, r = divmod(sum(A), 4) if r != 0: print('0') return q1, q2, q3 = q, q * 2, q * 3 q1_count = q2_count = q3_count = 0 s = 0 for a_i in A[:-1]: s += a_i if s == q3: q3_count += q2_count if s == q2: q2_count += q1_count if s == q1: q1_count += 1 print(q3_count) if __name__ == '__main__': main() {{tag>BOJ ps:problems:boj:골드_3}}