whu 1603 Minimum Sum(16武汉邀请赛网络赛)

最新推荐文章于 2017-04-17 15:17:27 发布

原创最新推荐文章于 2017-04-17 15:17:27 发布 · 354 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#暴力

其他暴力枚举专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种高效算法来解决从已排序数列中选取连续m个数，使这些数之间的绝对差之和达到最小的问题。通过巧妙利用相邻状态间的差异，将时间复杂度降低至O(n)，并提供了完整的实现代码。

题目链接：点这里！！

题意：给你n个数A[1] , A[2] .... A[n],从中选m个数 A[B[1]] , A[B[2]] ... A[B[m]] ( 1 <= B[1] < B[2] .... B[m] <= n )

使得sum abs( A[B[i]]-A[B[j]] ) when 1 <= i < j <= m. 尽可能的小。求sum的最小值。

数据范围：1<=m<=n<=100000 0<=A[i]<=100000

题解：易知，我们先给n个数排个序，然后就等价于从这个n个数选m个连续的数使得sum值最小。

我们直接一个个枚举过去就可以了(想太多)。

我们看第二组数据(已经排好序)：

5 3

1 3 6 8 10

我们枚举

1 3 6 得到的答案是10。

3 6 8 得到的答案是10。

6 8 10 得到的答案是8。

取最小值为8。

但是如果直接枚举过去的话，时间复杂度得到了O(n*n)，直接TLE了。

但我们看看相邻两个情况的区别

Ai,Ai+1, Ai+2,...,Ai+k-1 (1)

Ai+1,Ai+2,Ai+3...Ai+k (2)

(1)的答案为 (1-k)*Ai+(3-k)*(Ai+1)+(5-k)*(Ai+2)+...+(k-1)*(Ai+k-1) = sum1

(2)的答案为 (1-k)*(Ai+1)+(3-k)*(Ai+2)+...+(k-1)(Ai+k) = sum2

设dis[i]=A[1]+A[2]+...+A[i]

可以得出sum2 = sum1-(1-k)*Ai+(k-1)*(Ai+k) -2*(dis[i+k-1]-dis[i])

从而直接可推出下一个状态的答案。时间复杂度为O(n)

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<sstream>
#include<algorithm>
#include<vector>
#include<bitset>
#include<set>
#include<queue>
#include<stack>
#include<map>
#include<cstdlib>
#include<cmath>
#define PI 2*asin(1.0)
#define LL long long
#define pb push_back
#define pa pair<int,int>
#define clr(a,b) memset(a,b,sizeof(a))
#define lson lr<<1,l,mid
#define rson lr<<1|1,mid+1,r
#define bug(x) printf("%d++++++++++++++++++++%d\n",x,x)
#define key_value ch[ch[root][1]][0]
const int  MOD = 1000000007;
const int N = 1e5+1000;
const int maxn = 450;
const int letter = 130;
const int INF = 1e17;
const double pi=acos(-1.0);
const double eps=1e-8;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,k;
LL a[N],dis[N];
int main(){
    while(scanf("%d%d",&n,&k)!=EOF){
        for(int i=0;i<n;i++) scanf("%lld",a+i);
        sort(a,a+n);
        dis[0]=a[0];
        for(int i=1;i<n;i++) dis[i]=dis[i-1]+a[i];
        LL min1,sum=0;
        LL cnt=1ll*(1-k);
        for(int i=0;i<k;i++){
            sum += a[i]*cnt;
            cnt+=2;
        }
        min1=sum;
        int l=0;
        cnt=1ll*(1-k);
        for(int i=k;i<n;i++){
            sum -= cnt*a[l];
            sum += (-cnt)*a[i];
            sum += -2ll*(dis[i-1]-dis[l]);
            min1 =min(min1,sum);
            l++;
        }
        printf("%lld\n",min1);
    }
    return 0;
}