【STL—set】子段和

题目描述

给定一个序列，问你有没有可能重新排列这个序列使得所有长度小于等于 2 的子区间和不等于0。

输出描述:

一行一个字符串。"YES"表示可能重新排列这个序列使得所有长度小于等于
2
2 的子区间和不等于
0
0，反之则输出“NO”。输出不含引号。

示例1

输入
6
34 1345 -3542 -1423 4213 1

输出
YES

说明
不变即可。

示例2

输入

4
1 1 -1 1
输出

NO

说明
无论怎样排列，−1 总和至少 1 个 -1相邻，所以总存在一个和为 0长度为的子段。

示例3

输入

5
1 5 -1 -5 0
输出

NO
说明
0 的存在使得总有一个长度为 1的子段和为0

链接

牛客小白月赛46—B—子段和
【总结】C++ 基础数据结构 —— STL之集合（set）用法详解
for(auto i : v)遍历容器元素

代码

#include<bits/stdc++.h>
using namespace std;

const int N = 1e6 + 10;
set<int>a, b;

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++) {
        int t;
        cin >> t;
        if (t == 0) {
            cout << "NO"; return 0;
        }
        if (t > 0) {
            a.insert(t);
        }
        else {
            b.insert(t);
        }
    }
    if (b.empty()) {
        cout << "YES"; return 0;
    }
    else if (a.empty()) {
        cout << "YES"; return 0;
    }
    //遍历容器元素
    for (auto i : a) {
        for (auto j : b) {
            if (i + j != 0) {
                cout << "YES";
                return 0;
            }
        }
    }
    cout << "NO";
    return 0;
}