论今日水题试

有欧阳师者，一笑且考之

曰：日也长也,noipcsps日近矣

年年岁岁，算法尽学

此皆模板，一“易”字足以盖之

呜呼哉，LL不学新算法已多年

奈何欧阳教屡屡考以玄学

然则模板之易怎敌未学之难

哀耶，叹耶，

立而望之，不如灼灼新学

[题目描述]

给定一个长度为N的正整数序列{ }，设数对（i，j）的得分Sij=|ai-aj|，你需要找出K对数，使得这K对数的得分之和最小。

[输入格式]

第一行有两个正整数N，K，如题所述。

接下来一行有N个正整数，表示序列中的数。

[输出格式]

只有一个数，表示最小的得分之和。

[样例输入]

5 5
5 3 1 4 2

[样例输出]

[样例解释]

以下括号里的数表示下标。

选择（1，4），（2，4），（2，5），（3，5）这4对数，得分之和为4。

再从（1，2），（2，3），（4，5）中任选1对，因为这3对的得分都一样，都为2。

选择的数对（i，j）中i，j不能相等，且（i，j），（j，i）表示同一个数对，也就是说只能选其中一个。

[数据范围]

数据编号	N	K
1	10	40
2~4	2000	≤
5~7	100000	≤ 100000
8	500000	500000
9	100000	≤
10	1000000	≤

对于100%的数据：ai≤

#include<bits/stdc++.h>
#define re return
#define ll long long
#define inc(i,l,r) for(ll i=l;i<=r;++i)
using namespace std;
template<typename T>inline void rd(T&x)
{
    char c;bool f=0;
    while((c=getchar())<'0'||c>'9')if(c=='-')f=1;
    x=c^48;
    while((c=getchar())>='0'&&c<='9')x=x*10+(c^48);
    if(f)x=-x;
}

const int maxn=1000005;
ll n,K,a[maxn],sum[maxn];

inline bool check(ll jd)
{
    ll j=1,cnt=0;
    inc(i,2,n)
    {
        while(a[i]-a[j]>jd)++j;
        cnt+=i-j;
        if(cnt>=K)re 1;
    }
    re 0;
}

int main()
{
    //freopen("in.txt","r",stdin);
    rd(n),rd(K);
    inc(i,1,n)
    rd(a[i]);
    
    sort(a+1,a+n+1);
    ll l=0,r=a[n]-a[1];
    while(l<=r)
    {
        ll mid=(l+r)>>1;
        if(check(mid))r=mid-1;
        else l=mid+1;
    }

    ll ans=0;
    ll j=1,cnt=0;
    sum[1]=a[1];
    inc(i,2,n)
    {
        while(a[i]-a[j]>l)++j;
        cnt+=i-j;
        ans+=a[i]*(i-j)-sum[i-1]+sum[j-1]; 
        sum[i]=sum[i-1]+a[i];
    }
    
    if(cnt>K)ans-=(cnt-K)*l;
    printf("%lld",ans);
    re 0;
}

View Code