51nod 1267 4个数和为0(STL,二分)

题目链接: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 0x3f3f3f3f

const 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;
}



阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/mr_zj_/article/details/69525814
文章标签: STL 二分
个人分类: 二分 STL hash
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭