1090 3个数和为0

1090 3个数和为0

基准时间限制：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

题目不错,可惜我有点傻,一开始准备暴力,结果写到一半发现行不通,于是转为二分,
然而二分出现了一个很小的失误,然后连改了好几次,才发现.

#include <bits/stdc++.h>
#define N 1005
using namespace std;
int k[N],n;
bool Find(int a){
  int begin=0,end=n-1;
  while(begin<=end){
    int mid=(begin+end)>>1;
    if(k[mid]>a){
      end=mid-1;
    }else if(k[mid]<a){
      begin=mid+1;
    }else{
      return true;
    }
  }
  return false;
}
int main(){
  scanf("%d",&n);
  for(int i=0;i<n;i++){
    scanf("%d",&k[i]);
  }
  sort(k,k+n);
  bool prime=true;
  for(int i=0;i<n;i++)
    for(int j=i+1;j<n;j++){
      int ans=-(k[i]+k[j]);
      if(ans<=k[j])
          break;
      if(Find(ans)){
        prime=false;
        printf("%d %d %d
",k[i],k[j],ans);
      }
    }
  if(prime)
    printf("No Solution
");
  return 0;
}