CF-contest371E Subway Innovation
http://codeforces.com/contest/371/problem/E
题意:
直线上n个点,设i、j的距离为|Xi - Xj| ,求保留k个点,使得这些点两两之间的距离和最小。(保证无重点,读入非有序)
输出k个点的序号,若有多组则任意.
数据范围:
3<=n<=3e5 ,-1e8<= Xi <= 1e8 , 2<=k<= n-1
简单思路:
1. 排序之后,证明一下必须是连续的k个点才有可能满足。
2. 滑动窗口,大小为k。
3. 预处理出前缀和 或者 区间和,每次增加对窗口内和的影响为(k-1)*Xi - sum[i-k+1..i-1] ,删除同理。
代码:
#include <iostream>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <iomanip>
#include <cstdio>
using namespace std;
typedef long long LL;
const LL N = 300007;
const LL INF = 0x3f3f3f3f;
const double Pi = acos(-1);
LL n,m,k;
LL cc;
struct nd
{
LL x,id;
}s[N];
LL sum[N];
LL cmp(nd a,nd b)
{
return a.x<b.x;
}
int main()
{
LL cas=1;
while (~scanf("%I64d",&n))
{
for (LL i=0;i<n;i++)
{
scanf("%I64d",&s[i].x);
s[i].id=i+1;
}
scanf("%I64d",&k);
sort(s,s+n,cmp);
LL tmp=0;
LL l=0,r=k-1;
sum[0]=s[0].x;
for (LL i=1;i<n;i++)
sum[i]=sum[i-1]+s[i].x;
for (LL i=1;i<k;i++)
tmp+=i*s[i].x-sum[i-1];
LL mi=tmp;
for (LL i=k;i<n;i++)
{
LL t=i-k;
tmp-=sum[i-1]-sum[t]-(k-1)*s[t].x;
tmp+=(k-1)*s[i].x-(sum[i-1]-sum[t]);
if (tmp<mi)
{
mi=tmp;
r=i;
}
}
l=r-k+1;
for (LL i=l;i<=r;i++)
{
printf("%I64d",s[i].id);
if (i<r)
printf(" ");
else
printf("\n");
}
}
return 0;
}
/*
3
1 100 101
2
5
11 21 30 40 50
3
*/