poj 2785 二分的方法

来源：互联网发布：三星淘宝旗舰店编辑：程序博客网时间：2024/05/24 07:42

这个题的大题意思就是需要从四列中随机找四个数，他们的和为零，四重循环是不可能的，想了想分成两组去做，发现也超时了，后来就用了二分的方法过的。

也是挺头疼的，也学习一波二分。

#include<stdio.h>#include<string.h>#include<algorithm>#include <iostream>using namespace std;int a[4][4444],n;int num1[16100000],num2[16100000],c;int erfen(int left,int right,int k){    int i;    while(left<=right)    {        int mid=(left+right)/2;        int num=0;        if(num2[mid]==k)        {            num=1;            for(i=mid-1; i>=0&&num2[i]==k; i--)  num++;            for(i=mid+1; i<n*n&&num2[i]==k; i++)  num++;            return num;        }        else if(num2[mid]>k)            right=mid-1;        else left=mid+1;    }    return 0;}int main(){    int i,j;    while(scanf("%d",&n)!=EOF)    {        int num=0;        c=0;        for(i=0; i<n; i++)        {            scanf("%d %d %d %d",&a[0][i],&a[1][i],&a[2][i],&a[3][i]);        }        for(i=0; i<n; i++)            for(j=0; j<n; j++)            {                num1[c]=a[2][i]+a[3][j];                num2[c++]=-(a[0][i]+a[1][j]);            }        sort(num2,num2+c);        for(i=0; i<c; i++)        {            num+=erfen(0,n*n-1,num1[i]);        }        printf("%d\n",num);    }    return 0;}

阅读全文

1 0