51nod 1267 4个数和为0（STL，二分）

来源：互联网发布：c语言经典笔试题编辑：程序博客网时间：2024/05/18 03:51

题目链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1267

解题方案：4个数a,b,c,d的和可以拆成(a+b)+(c+d)的和，即可以先预处理出所有可以凑成的2个数的和，然后在这所有的2个数的和里面进行二分查找0-(a+b)。因为题目没有说N个数互不相同，所以二分找到以后还得check一下是否共用了某个数，如果共用了的话就得看这个数的个数是否够用。复杂度O(1e6*log(le6))。

#include <iostream>#include <cstdio>#include <algorithm>#include <cstring>#include <string>#include <cmath>#include <vector>#include <queue>#include <stack>#include <set>#include <map>#include <time.h>using namespace std;#define FOR(i,k,n) for(int i=k;i<n;i++)#define FORR(i,k,n) for(int i=k;i<=n;i++)#define scan(a) scanf("%d",&a)#define scann(a,b) scanf("%d%d",&a,&b)#define scannn(a,b,c) scanf("%d%d%d",&a,&b,&c)#define mst(a,n)  memset(a,n,sizeof(a))#define ll long long#define N 1005#define mod 1000000007#define INF 0x3f3f3f3fconst double eps=1e-8;const double pi=acos(-1.0);int a[N];map<int,int> num;typedef map<int,vector<pair<int,int> > > Sum;Sum sum;int main(){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    int n;    cin>>n;    FOR(i,0,n) cin>>a[i],num[a[i]]++;    //FOR(i,0,n) printf("%d\n",a[i]);    sort(a,a+n);    FOR(i,0,n)    {        FOR(j,i+1,n)        {            sum[a[i]+a[j]].push_back({a[i],a[j]});        }    }    for(Sum::iterator it=sum.begin();it!=sum.end();it++)    {        Sum::iterator it1=sum.find(0-(it->first));        if(it1!=sum.end())        {            vector<pair<int,int> > v1,v2;            v1=it->second; v2=it1->second;            int a,b,c,d;            FOR(i,0,v1.size())            {                a=v1[i].first; b=v1[i].second;                num[a]--; num[b]--;                FOR(j,0,v2.size())                {                    c=v2[j].first; d=v2[j].second;                    num[c]--; num[d]--;                    if(num[a]>=0&&num[b]>=0&&num[c]>=0&&num[d]>=0)                    {                        //printf("%d %d %d %d\n",a,b,c,d);                        puts("Yes");                        return 0;                    }                    num[c]++; num[d]++;                }                num[a]++; num[b]++;            }        }    }    puts("No");    return 0;}

0 0