python数据结构与算法练习-方差
方差
链接: link.
input
5 3
1 2 3 4 5
output
6
##思路:
- 使用itertools 中的combations枚举n中所有m的排列
- 分别计算每个排列的结果然后返回最小值
(本人思路最后的时间复杂度是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);
}
官网有更优解法,这里仅供自己复习使用,也欢迎批评指正。