基准时间限制:1 秒 空间限制:131072 KB 分值: 5
难度:1级算法题
给出一个长度为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
刚开始用了搜索,不断剪枝,还是有三个数据TLE。应该是有的数据任意三个数组合都不为零,这样的话遍历完太费时间。
看了网上的思路,每次找两个,二分查找第三个数
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
ll mat[1005];
ll a[5];
ll vis[1005];
int n;
int flag;
int main()
{
while(~scanf("%d",&n))
{
for(int i=0;i<n;i++)
scanf("%lld",&mat[i]);
sort(mat,mat+n);
flag=0;
for(int i=0;i<n-1;i++)
{
for(int j=i+1;j<n;j++)
{
if(*lower_bound(mat+j+1,mat+n,-mat[i]-mat[j])==-mat[i]-mat[j])
{
printf("%lld %lld %lld\n",mat[i],mat[j],-mat[i]-mat[j]);
flag=1;
}
}
}
if(!flag) printf("No Solution\n");
}
return 0;
}