Wannafly挑战赛21 C 大水题

https://www.nowcoder.com/acm/contest/159/C

dp

#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <list>
#include <stack>
#include <queue>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
#define ll long long
#define minv 1e-6
#define inf 1e9
const ll mod=1e9+;
const long maxn=3e5+;

struct node
{
    int a,num;
}p[maxn];

int pre[maxn];
ll g[maxn],tot[maxn],f[maxn][];

int cmp(node x,node y)
{
    if (x.a==y.a)
        return x.num<y.num;
    else
        return x.a<y.a;
}

int main()
{
    int n,i;
    ll maxf=;
    scanf("%d",&n);
    for (i=;i<=n;i++)
    {
        scanf("%d",&p[i].a);
        p[i].num=i;
    }
    sort(p+,p+n+,cmp);

    for (i=;i<=n;i++)
        pre[i]=-;
    for (i=;i<=n;i++)
        if (p[i].a==p[i-].a)
            pre[p[i].num]=p[i-].num;

    tot[]=;
    for (i=;i<=n;i++)
    {
        scanf("%lld",&g[i]);
        tot[i]=tot[i-]+g[i];
    }

    maxf=;
    f[][]=,f[][]=;
    for (i=;i<=n;i++)
    {
        f[i][]=max(f[i-][],f[i-][]);
        if (pre[i]!=-)
            f[i][]=max(f[pre[i]][]+tot[i]-tot[pre[i]-] , f[pre[i]][]-g[pre[i]]+tot[i]-tot[pre[i]-]);
        else
            f[i][]=;
    }
    printf("%lld",max(f[n][],f[n][]));

//    maxf=0;
//    for (i=1;i<=n;i++)
//    {
//        f[i]=f[i-1];
//        if (pre[i]!=-1)
//            f[i]=max(f[i-1],f[pre[i]]+tot[i]-tot[pre[i]-1]);
//
//    }
//    printf("%lld",f[n]);

    return ;
}
/*
7
1 1 1 2 1 1 1
1 2 3 1 4 5 6

7
1 2 1 4 7 1 7
1 1 1 1 1 1 10
*/