UVa1152 4 Values whose Sum is 0 (中途相遇法+二分)

题目链接:https://vjudge.net/problem/UVA-1152

题目大意:你输入四排数,每排代表一个集合,你可以从每个集合里边取出一个数,让其加和为0。
题目分析:这个题目最简单的做法就是n^4四个循环进行求解,但是这样就会爆时间,这样我们就可以采用中途相遇法,就是把a+b+c+d分解为a+b和c+d,然后再对a+b排序,再通过枚举c+d然后二分找到a+b里边符合题意的。(网上还有hash的帖子,这样可能速度会更快)

学到的东西:又熟悉了一遍二分的题目,了解了中间相遇法,多维问题可以把它分解为一个个低维的小问题。

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=16e6+10;
int m[4010][5];
int ab[maxn];
int cd[maxn];
int main()
{
    int T;scanf("%d",&T);while(T--){
        int n;scanf("%d",&n);
        for(int i=0;i<n;i++)
            for(int j=0;j<4;j++)
            scanf("%d",&m[i][j]);

        for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){//中间相遇法
            ab[i*n+j]=m[i][0]+m[j][1];
            cd[i*n+j]=m[i][2]+m[j][3];
        }

        int ans=0;
        sort(ab,ab+n*n);
        for(int i=0;i<n*n;i++){//枚举cd 找到对应的ab
            int x=(-cd[i]);
            int l=0,r=n*n-1;
            while(l<=r){
                int mid=(l+r)>>1;
                if(ab[mid]>x) r=mid-1;
                else if(ab[mid]<x) l=mid+1;
                else{       //二分找到合适的
                    for(int k=mid;k>=0;k--){  //采用这种方法,要快于O n的做法
                        if(ab[k]==x) ans++;
                        else break;
                    }
                    for(int k=mid+1;k<=r;k++){
                        if(ab[k]==x) ans++;
                        else break;
                    }
                    break;
                }
            }
        }
        printf("%d\n",ans);
        if(T) printf("\n");
    }
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值