UVA-1152

把a+b与c+d的结果储存到两个数组里面,再排序,“快排”一样从头到尾和从尾到头遍历。注意输出不要PE;

#include<bits/stdc++.h>
using namespace std;
//-1 -1 1 1
int main(void){
    int n;
    cin>>n;
    while(n--){
        int l;
        scanf("%d",&l);
        int a[l],b[l],c[l],d[l];
        for(int i=0;i<l;i++){
            scanf("%d %d %d %d",&a[i],&b[i],&c[i],&d[i]);
        }
        int ll=l*l;
        int ab[ll],cd[ll];
        for(int i=0;i<l;i++){
            for(int j=0;j<l;j++){
                ab[i*l+j]=a[i]+b[j];
                cd[i*l+j]=c[i]+d[j];
            }
        }
        sort(ab,ab+ll);
        sort(cd,cd+ll);
//        for(int i=0;i<ll;i++){
//            printf("%d ",ab[i]);
//        }
//        for(int i=0;i<ll;i++){
//            printf("%d ",cd[i]);
//        }
        int ct=0;
        int p=0,q=ll-1;
        while(p<ll&&q>-1){
            if(ab[p]+cd[q]==0){
                int t1=p,t2=q;
                while(1){
                    t1++;
//                    printf("%d\n",t1);
                    if(t1==ll||ab[t1]!=ab[p]){
                        break;
                    }
                }
//                printf("abt1=%d,abp=%d\n",ab[t1],ab[p]);
                while(t2--){
                    if(t2==-1||cd[t2]!=cd[q]){
                        break;
                    }
                }
//                printf("t1=%d,t2=%d,ll=%d\n",t1,t2,ll);
                ct+=(t1-p)*(q-t2);
                p=t1;
                q=t2;
            }
            else if(ab[p]+cd[q]>0){
                q--;
            }
            else {
                p++;
            }
//            printf("ct=%d\n",ct);
        }
        cout<<ct<<endl;
        if(n){
            printf("\n");
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值