|
| 1 | +#!usr/bin/env python3 |
| 2 | +from collections import defaultdict, deque, Counter, OrderedDict |
| 3 | +from bisect import bisect_left, bisect_right |
| 4 | +from functools import reduce, lru_cache |
| 5 | +from heapq import heappush, heappop, heapify |
| 6 | + |
| 7 | +from itertools import * |
| 8 | +import math |
| 9 | +import fractions |
| 10 | +import sys |
| 11 | +import copy |
| 12 | + |
| 13 | + |
| 14 | +def L(): return sys.stdin.readline().split() |
| 15 | +def I(): return int(sys.stdin.readline().rstrip()) |
| 16 | +def SL(): return list(sys.stdin.readline().rstrip()) |
| 17 | +def LI(): return [int(x) for x in sys.stdin.readline().split()] |
| 18 | +def LI1(): return [int(x) - 1 for x in sys.stdin.readline().split()] |
| 19 | +def LS(): return [list(x) for x in sys.stdin.readline().split()] |
| 20 | +def R(n): return [sys.stdin.readline().strip() for _ in range(n)] |
| 21 | +def LR(n): return [L() for _ in range(n)] |
| 22 | +def IR(n): return [I() for _ in range(n)] |
| 23 | +def LIR(n): return [LI() for _ in range(n)] |
| 24 | +def LIR1(n): return [LI1() for _ in range(n)] |
| 25 | +def SLR(n): return [SL() for _ in range(n)] |
| 26 | +def LSR(n): return [LS() for _ in range(n)] |
| 27 | + |
| 28 | + |
| 29 | +def perm(n, r): return math.factorial(n) // math.factorial(r) |
| 30 | + |
| 31 | + |
| 32 | +def comb(n, r): return math.factorial( |
| 33 | + n) // (math.factorial(r) * math.factorial(n-r)) |
| 34 | + |
| 35 | + |
| 36 | +def powerset(iterable): |
| 37 | + s = list(iterable) |
| 38 | + return chain.from_iterable(combinations(s, r) for r in range(len(s)+1)) |
| 39 | + |
| 40 | + |
| 41 | +def make_list(n, *args, default=0): return [make_list(*args, default=default) |
| 42 | + for _ in range(n)] if args else [default for _ in range(n)] |
| 43 | + |
| 44 | + |
| 45 | +dire = [[1, 0], [0, 1], [-1, 0], [0, -1]] |
| 46 | +dire8 = [[1, 0], [1, 1], [0, 1], [-1, 1], [-1, 0], [-1, -1], [0, -1], [1, -1]] |
| 47 | +alphabets = "abcdefghijklmnopqrstuvwxyz" |
| 48 | +ALPHABETS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" |
| 49 | +MOD = 1000000007 |
| 50 | +INF = float("inf") |
| 51 | + |
| 52 | +sys.setrecursionlimit(1000000) |
| 53 | + |
| 54 | + |
| 55 | +def main(): |
| 56 | + N = I() |
| 57 | + A = LI() |
| 58 | + |
| 59 | + cnt = [0] * 401 |
| 60 | + for a in A: |
| 61 | + cnt[a+200] += 1 |
| 62 | + |
| 63 | + ans = 0 |
| 64 | + for i, c1 in enumerate(cnt): |
| 65 | + for j, c2 in enumerate(cnt): |
| 66 | + ans += c1*c2*(((i-200) - (j-200))**2) |
| 67 | + print(ans//2) |
| 68 | + |
| 69 | + |
| 70 | +if __name__ == '__main__': |
| 71 | + main() |
0 commit comments