3743. Alice and Bob

题目描述

Sample Input
输入1：

4

1 2 2 3

输入2：

4

1 1 2 3

Sample Output
输出1：

5

输出2：

5

N<=10^5

题解

注意给出的是a

一种神奇的做法

从后往前求max(b[i])，假设b[i+1]~b[n]已经求出，考虑能转移到b[i]的b[j]

设pre[i]表示最后一个a[i]-1的位置（a[0]=0），把(i,a[i])丢到笛卡尔坐标系上，从(i,a[i])向(pre[i],a[pre[i]])连线，则图形成一棵树

若b[j]能转移到b[i]，那么i向下做一条不经i点的射线，把射线经过的边断开，则必然能把j点与0点分开

如图，则b[2]可以由b[3]和b[4]转移，但b[3]不能由b[4]转移

证明：

线的实质就是x=a时a数组的转移情况，现要使b最大，那么就要把x尽量上移

如果x[i]经过了一条线，则会导致a[i]改变，不满足要求

---

每次可以把一个点以及其后面的点的答案向pre传，同时维护以a为关键字的线段树即可

code

#include <bits/stdc++.h>
#define fo(a,b,c) for (a=b; a<=c; a++)
#define fd(a,b,c) for (a=b; a>=c; a--)
#define max(a,b) (a>b?a:b)
#define min(a,b) (a<b?a:b)
#define fout
using namespace std;

int tr[500002];
int a[100002];
int f[100002];
int f2[100002];
int ls[100002];
int pre[100002];
int n,i,j,k,l;
long long ans;

void change(int t,int l,int r,int x,int s)
{
    int mid=(l+r)/2;
    
    tr[t]=max(tr[t],s);
    if (l==r) return;
    
    if (x<=mid)
    change(t*2,l,mid,x,s);
    else
    change(t*2+1,mid+1,r,x,s);
}

int find(int t,int l,int r,int x,int y)
{
    int mid=(l+r)/2,s,ans=0;
    
    if (x<=l && r<=y)
    return tr[t];
    
    if (x<=mid)
    s=find(t*2,l,mid,x,y),ans=max(ans,s);
    if (mid<y)
    s=find(t*2+1,mid+1,r,x,y),ans=max(ans,s);
    
    return ans;
}

int main()
{
    freopen("alice.in","r",stdin);
    #ifdef fout
    freopen("alice.out","w",stdout);
    #endif
    
    scanf("%d",&n);
    fo(i,1,n)
    scanf("%d",&a[i]);
    
    fo(i,1,n)
    {
        pre[i]=ls[a[i]-1];
        ls[a[i]]=i;
    }
    
    fd(i,n,1)
    {
        f[i]=find(1,0,n,0,a[i]-1)+1;
        
        f2[i]=max(f2[i],f[i]);
        f2[pre[i]]=max(f2[pre[i]],f2[i]);
        
        change(1,0,n,a[i],f2[i]);
        change(1,0,n,a[pre[i]],f2[pre[i]]);
    }
    
    fo(i,1,n)
    ans+=f[i];
    
    printf("%lld\n",ans);
    
    fclose(stdin);
    fclose(stdout);
    
    return 0;
}