3782 点（推公式）

1. 问题描述：

给定一个平面上的 N 个点，请你计算所有点对之间距离的平方和。

输入格式

第一行包含整数 N，表示点的个数。接下来 N 行，每行包含两个整数 X，Y，表示一个点的横纵坐标。注意，不保证所有点的位置各不相同。

输出格式

输出所有点对之间距离的平方和。

数据范围

1 ≤ N ≤ 10 ^ 5，
−10000 ≤ X，Y ≤ 10000。

输入样例：

4
1 1
-1 -1
1 -1
-1 1

输出样例：

32
来源：https://www.acwing.com/problem/content/description/3785/

2. 思路分析：

因为数据规模比较大所以不能够直接暴力枚举，暴力枚举的时间复杂度为O(n ^ 2)肯定会超时的，所以我们需要推导一下具体的公式，也即先化简一下：

 

最终每一项可以使用O(n)的时间内计算出结果，并且x和y是相互独立的所以可以分别计算出对应的结果相加即可。

3. 代码如下：

from typing import List


class Solution:
    # 获取(xi - xj) ^ 2, 其中i < j
    def get(self, n: int, x: List[int]):
        # s1和s2分别存储平方和与所有数字之和
        s1 = s2 = 0
        for i in range(n):
            s1 += x[i] * x[i]
            s2 += x[i]
        return n * s1 - s2 * s2

    def process(self):
        n = int(input())
        # x, y分别存储横坐标与纵坐标
        x, y = list(), list()
        for i in range(n):
            a, b = map(int, input().split())
            x.append(a)
            y.append(b)
        # x与y是独立的所以可以分开计算
        return self.get(n, x) + self.get(n, y)


if __name__ == '__main__':
    print(Solution().process())