51nod 1090 & 1267 【二分简单题】

最新推荐文章于 2021-08-30 18:17:13 发布

转载最新推荐文章于 2021-08-30 18:17:13 发布 · 113 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/bestwzh/p/6636963.html

本文介绍了一种寻找数组中三个数相加等于特定值的有效算法。通过先排序再使用双指针技巧与二分查找结合的方法，快速定位符合条件的三元组。此算法能够高效解决三数求和问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

做法：从左往右枚举前两个数的和sum，剩余的数二分找-sum是否存在。

#include <bits/stdc++.h>
using namespace std;
struct Node {
    int a, b, c;
}temp;
int main() {
    int n;
    int a[1010];
    scanf("%d", &n);
    for(int i = 0; i < n; i++) scanf("%d", &a[i]);
    sort(a, a+n);
    vector<Node>ans;
    for(int i = 0; i < n-2; i++) {
        for(int j = i+1; j < n-1; j++) {
            int sum = a[i] + a[j];
            int pos = lower_bound(a+j+1, a+n, -sum) - a;
            if(pos >= n || a[pos] != -sum) continue;
//            cout << i << ' ' << j <<' ' << pos << endl;
            temp.a = a[i];
            temp.b = a[j];
            temp.c = a[pos];
            ans.push_back(temp);
        }
    }
    if(ans.size() == 0) {
        puts("No Solution");
        return 0;
    }
    for(int i = 0; i < ans.size(); i++) {
        cout << ans[i].a << ' ' << ans[i].b << ' ' << ans[i].c << endl;
    }
}

感觉比第一题还简单，是因为数据太弱？三个for枚举前三个数和sum, 二分剩余的数找-sum。

测试数据三个数sum居然不会爆int

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n;
    int a[1010];
    scanf("%d", &n);
    for(int i = 0; i < n; i++) scanf("%d", &a[i]);
    sort(a, a+n);
    for(int i = 0; i < n-3; i++) {
        for(int j = i+1; j < n-2; j++) {
            for(int k = j+1; k < n-1; k++) {
                int sum = a[i] + a[j]; sum += a[k];
                int pos = lower_bound(a+k+1, a+n, -sum) - a;
                if(pos >= n || a[pos] != -sum) continue;
                puts("Yes");
                return 0;
            }
        }
    }
    puts("No");
}