[51NOD1065] 最小正子段和（STL，前缀和）

题目链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1065

估计没人这么做吧…用一个set维护前缀和，但是set的lower_bound和upper_bound都是返回不小于x的最小值，这就很尴尬了。

 重载了这个<，在树里面前序遍历出来的就是降序了。 然后二分的时候还是用的这个<，原本意思是找到不小于的第一个，但是重载以后，查找的位置不变，结果就是不大于的了。

#include <bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;

typedef long long LL;
typedef struct F {
  LL v;
  F() {}
  F(LL v) : v(v) {}
  bool operator<(const F &x) const {
    return v > x.v;
  }
}F;
const int maxn = 50050 * 2;
int n;
int a[maxn];
LL s[maxn];
set<F> v;

int main() {
  //freopen("in", "r", stdin);
  while(~scanf("%d", &n)) {
    memset(s, 0, sizeof(s));
    v.clear(); v.insert(s[0]);
    LL ret = (LL)1 << 60;
    for(int i = 1; i <= n; i++) {
      scanf("%d", &a[i]);
      s[i] = s[i-1] + a[i];
      if(a[i] >= 1) ret = min(ret, (LL)a[i]);
      if(s[i] >= 1) ret = min(ret, s[i]);
      F tmp = *v.lower_bound(s[i]);
      if(s[i] - tmp.v >= 1) {
        ret = min(ret, s[i]-tmp.v);
      }
      v.insert(F(s[i]));
    }
    cout << ret << endl;
  }
  return 0;
}

噗，直接在set后面加一个greater就好。。

#include <bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;

typedef long long LL;
const int maxn = 50050 * 2;
int n;
int a[maxn];
LL s[maxn];
set<LL, greater<LL> > v;

int main() {
  //freopen("in", "r", stdin);
  while(~scanf("%d", &n)) {
    memset(s, 0, sizeof(s));
    v.clear(); v.insert(s[0]);
    LL ret = (LL)1 << 60;
    for(int i = 1; i <= n; i++) {
      scanf("%d", &a[i]);
      s[i] = s[i-1] + a[i];
      if(a[i] >= 1) ret = min(ret, (LL)a[i]);
      if(s[i] >= 1) ret = min(ret, s[i]);
      LL tmp = *v.upper_bound(s[i]);
      if(s[i]-tmp >= 1) {
        ret = min(ret, s[i]-tmp);
      }
      v.insert(s[i]);
    }
    cout << ret << endl;
  }
  return 0;
}