zznuoj-2060 Minsum Plus

[题目链接]

刚开始没想出来用贪心写，用set维护的前缀和，无奈有个变量用的int，一直WA…

思路1：用sum数组记录前缀和，从后往前遍历sum数组，假设当前位置为i，用upper_bound(sum[i])找到比sum[i]大的最小值x，不断更新Min=min(Min,x-sum[i])即可。

代码：

#include<iostream>
#include<cstdio>
#include<set>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll inf=0xffffffff;
const int Max_n=5e4+10;

int n;
ll a[Max_n];
ll sum[Max_n];

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
    sum[0]=0;
    for(int i=1;i<=n;i++)sum[i]=sum[i-1]+a[i];
    ll Min=inf;
    set<ll>s;
    for(int i=n;i>=0;i--){
        set<ll>::iterator it=s.upper_bound(sum[i]);
        if(it!=s.end()&&*it-sum[i]<Min)
            Min=*it-sum[i];
        s.insert(sum[i]);
    }
    if(Min==inf)printf("no solution\n");
    else printf("%lld\n",Min);
    return 0;
}

思路2：如果ABC为排序后的结果，那么当A和B不能组成序列，而A和C可以组成序列时，那么B和C一定可以组成序列，并且BC一定会比AC更优，Orz ~

代码：

#include<iostream>
#include<cstdio>
#include<set>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll inf=0xffffffff;
const int Max_n=5e4+10;

int n;
struct Node{
    int pos;
    ll sum;
    bool operator<(const Node& n)const{
        if(sum!=n.sum)return sum<n.sum;
        return pos>n.pos; //不可省 0 4 1 -1
    }
}node[Max_n];

int main()
{
    scanf("%d",&n);
    node[0].pos=0;node[0].sum=0;
    int t;
    for(int i=1;i<=n;i++){
        scanf("%d",&t);
        node[i].pos=i;
        node[i].sum=node[i-1].sum+t;
    }
    sort(node,node+n+1);
    ll Min=inf;
    for(int i=1;i<=n;i++){
        if(node[i].pos>node[i-1].pos&&node[i].sum-node[i-1].sum>0)
            Min=min(Min,node[i].sum-node[i-1].sum);
    }
    if(Min==inf)printf("no solution\n");
    else printf("%lld\n",Min);
    return 0;
}