CodeForces Round 218 E Subway Innovation

题目大意：

直线上N个点，去掉一些只剩下K个，使得剩下的K个点两两之间距离之和最小

显然，这K个点应该是坐标轴上连续的K个点

但是因为N很大，所以只能是O（n）的做

第一O(n)的求出 0 到 k 的和

第二O(n)的求出 剩下的长度为k的区间的答案

先求第一个

f(0,i) = f(0,i-1)+∑dist[i]-dist[k] (k<-[0,i-1])

      = f(0,i-1)+(i-1)*dist[i]-sum[0,i-1]; 

只要求出前缀和就可以O(1)的回答

同理求出从 ans[i,j]到ans[i+1,j+1]的公式，同样利用前缀和

#include<string.h>
#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;

#define ll long long 

typedef pair<ll,int > P;

#define fst first 
#define sec second

P dist[400000];
ll sum[400000];
ll ans[400000];

int main()
{
	int n,k;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		ll t;
		cin>>t;
		dist[i] = P(t,i);
	}
	cin>>k;
	sort(dist+1,dist+1+n);
	for(int i=1;i <= n ;i++)
		sum[i]=sum[i-1]+dist[i].fst;
	ll t=0;
	for(int i=1;i<=k;i++)
		t=t+(i-1)*dist[i].fst-sum[i-1];
	ll ans=t;
	int s=1;
	for(int i=k+1;i<=n;i++)
	{
		t=t-2*(sum[i-1]-sum[i-k+1]+dist[i-k+1].fst)+(k-1)*(dist[i].fst+dist[i-k].fst);
		if(t<ans)
		{
			s=i-k+1;
			ans=t;
		}
	}
	for(int i=0;i<k;i++)
	{
		cout<<dist[s+i].sec<<" ";
	}
	return 0;
}