数据结构与算法练习-方差（排序）

方差

链接: link.

input

5 3
1 2 3 4 5

output

6

##思路：

1. 使用itertools 中的combations枚举n中所有m的排列
2. 分别计算每个排列的结果然后返回最小值
   （本人思路最后的时间复杂度是O(n2)，想了一个多小时不知道怎么O（n）情况下做出来，这里贴一下网上别人的思路）

这里分别贴上我的笨解和网上的优解：
O(n2)：

from collections import deque
from itertools import combinations
n,m = map(int,input().split())
nums = list(map(int,input().split()))
def p_sum(nums,n):
    sum = 0
    for i in nums:
        sum += i
    return sum / n
    
def sqr_sum(l,a,m):
    sum = 0
    for i in l:
        sum = sum + abs(i-a)**2
    return sum / m
    
#这里注意一个知识点就是求出队列的所有排列组合 需要用到combinations函数 from itertools import combinations
def max_sqr(nums,m):
    maxs = deque()
    for i in combinations(nums,m):
        # li = list(i)
        s = sqr_sum(i,p_sum(i,m),m)
        maxs.append(s)
    return int(min(maxs)*m**2)
print(max_sqr(nums,m))

O（n）:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn=1e6+10;
ll a[maxn],sum1[maxn],sum2[maxn];
int main(){
    ll n,m;
    scanf("%lld %lld",&n,&m);
    for(ll i=1; i<=n; i++) scanf("%lld",&a[i]);
    sort(a+1,a+n+1);
    for(ll i=1; i<=n; i++){
        sum1[i]=sum1[i-1]+a[i];
        sum2[i]=sum2[i-1]+a[i]*a[i];
    }
    ll minn=LONG_LONG_MAX;
    for(ll i=m; i<=n; i++){
        ll tmp=(sum2[i]-sum2[i-m])*m-(sum1[i]-sum1[i-m])*(sum1[i]-sum1[i-m]);
        minn=min(minn,tmp);
    }
    printf("%lld\n",minn);
}

官网有更优解法，这里仅供自己复习使用，也欢迎批评指正。