基准时间限制:1 秒 空间限制:131072 KB 分值: 5 难度:1级算法题
51nod 1090 3个数和为0【二分】-LMLPHP 收藏
51nod 1090 3个数和为0【二分】-LMLPHP 关注
给出一个长度为N的无序数组,数组中的元素为整数,有正有负包括0,并互不相等。从中找出所有和 = 0的3个数的组合。如果没有这样的组合,输出No Solution。如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则按照第二小的数排序。

 
Input
第1行,1个数N,N为数组的长度(0 <= N <= 1000)
第2 - N + 1行:A[i](-10^9 <= A[i] <= 10^9)
Output
如果没有符合条件的组合,输出No Solution。
如果有多个,按照3个数中最小的数从小到大排序,如果最小的数相等则继续按照第二小的数排序。每行3个数,中间用空格分隔,并且这3个数按照从小到大的顺序排列。
Input示例
7
-3
-2
-1
0
1
2
3
Output示例
-3 0 3
-3 1 2
-2 -1 3
-2 0 2
-1 0 1 【分析】:直接枚举两个数,二分查找第三个数就行了。由于题中要求按数的大小排序输出,所以直接sort升序输出即可,这样即可用二分查找第三个数,即-a[i]-a[j],并且一定比a[i]和a[j]大,复杂度为n^2log(n);
【代码】:
#include <bits/stdc++.h>

using namespace std;

const int maxn = +;
int n;
int a[maxn]; int check(int x)
{
int l=,r=n-;
while(l<=r)
{
int mid=(l+r)>>;
if(a[mid]==x)
return ;
if(a[mid]<x)
l=mid+;
else
r=mid-;
}
return ; }
int main()
{
int ans;
int flag=; cin>>n;
for(int i=;i<n;i++)
cin>>a[i];
sort(a,a+n);
for(int i=;i<n;i++)
{
for(int j=i+;j<n;j++)
{
int ans=-a[i]-a[j];
if(ans<=a[j])
break;
if(check(ans))
{
cout<<a[i]<<" "<<a[j]<<" "<<ans<<endl;
flag=;
} }
}
if(!flag)
cout<<"No Solution"<<endl;
return ;
}

二分

#include <bits/stdc++.h>

using namespace std;
const int maxn = +;
int main()
{
int n;
int flag=;
int a[maxn];
cin>>n;
for(int i=;i<n;i++)
cin>>a[i];
sort(a,a+n);
for(int i=;i<n;i++)
{
for(int j=i+;j<n;j++)
{
if(a[i]+a[j]<=)
{
for(int k=j+;k<n;k++)
{
if(a[i]+a[j]+a[k]==)
{
cout<<a[i]<<" "<<a[j]<<" "<<a[k]<<endl;
flag=;
}
}
}
}
}
if(flag)
cout<<"No Solution"<<endl;
return ;
}

暴力枚举+剪枝

05-11 17:54