【jzoj5288】【NOIP2017提高组A组模拟8.17】【球场大佬】

Description

每天下午,古猴都会去打羽毛球。但是古猴实在是太强了,他必须要到一些比较强的场去打。但是每个羽毛球场都有许多的人排着队,每次都只能上四个人,每个人都有自己的能力值,然而这四个人的总能力的高低与否才是古猴是否决定参加这个场的关键。

每四个人的总能力值的定义是:任意选两个与另两个PK,能力值的贡献是较高的一组减去较低的一组。比如能力值为5和7的去PK 6和10的差值,那么用较高的减去较低的就是6+10-5-7=4。然后四个人的总能力值要任意两两之间与其他两个的总贡献。如(a,b,c,d)四个人,那么他们的总能力值就是(a,b)一组与(c,d)一组PK,(a,c)一组与(b,d)一组PK,(a,d)一组与(b,c)一组PK,(b,c)一组与(a,d)一组PK,(b,d)一组与(a,c)一组PK,(c,d)一组与(a,b)一组PK这六项PK差值就是四个人的总能力值。

现在,古猴想知道这个场任意四个人的总能力值的和是多少,但是急着要拿拍,你需要马上告诉他这个场的情况?

solution

可以先两两组合排序,n^2可以统计出两两匹配的贡献,但是有可能算重,组合i(a,b)和组合j(c,d),a有可能等于c,我们可以枚举b,d,算有多少种a==c,去重即可。

code

#include
#include
#include
#include
#include
#define LL long long
#define fo(i,j,k) for(LL i=j;i<=k;i++)
#define fd(i,j,k) for(LL i=j;i>=k;i--)
#define fr(i,j) for(LL i=begin[j];i;i=next[i])
using namespace std;
LL const mn=2000+9,mo=1e9+7;
LL t,n,a[mn],b[mn*mn];
int main(){
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%lld",&t);
    fo(cas,1,t){
        scanf("%lld",&n);
        fo(i,1,n)scanf("%lld",&a[i]);
        b[0]=0;
        LL tmp=0;
        fo(i,1,n)fo(j,i+1,n)b[++b[0]]=a[i]+a[j],tmp=(tmp+b[b[0]])%mo;
        sort(b+1,b+b[0]+1);
        LL ans=0,tm2=0;
        fo(i,1,b[0]){
            tmp=(tmp-b[i])%mo;
            ans=(ans+b[i]*(i-1)-tm2+tmp-b[i]*(b[0]-i))%mo;
            tm2=(tm2+b[i])%mo;
        }
        fo(i,1,n)fo(j,i+1,n)ans=(ans-abs(a[i]-a[j])*2*(n-2))%mo;
        printf("%lld\n",(ans+mo)%mo);
    }
    return 0;
}

你可能感兴趣的:(jzoj,数论)