洛谷 P4093 [HEOI2016/TJOI2016]序列 k-d tree

题目描述

佳媛姐姐过生日的时候，她的小伙伴从某宝上买了一个有趣的玩具送给他。玩具上有一个数列，数列中某些项的值可能会变化，但同一个时刻最多只有一个值发生变化。现在佳媛姐姐已经研究出了所有变化的可能性，她想请教你，能否选出一个子序列，使得在任意一种变化中，这个子序列都是不降的？请你告诉她这个子序列的最长长度即可 。

注意：每种变化最多只有一个值发生变化。在样例输入1中，所有的变化是：

1 2 3
2 2 3
1 3 3
1 1 3
1 2 4
选择子序列为原序列，即在任意一种变化中均为不降子序列在样例输入2中，所有的变化是:

3 3 3
3 2 3
选择子序列为第一个元素和第三个元素，或者第二个元素和第三个元素，均可满足要

输入输出格式

输入格式：
输入的第一行有两个正整数n, m，分别表示序列的长度和变化的个数。接下来一行有n个数，表示这个数列原始的状态。接下来m行，每行有2个数x, y，表示数列的第x项可以变化成y这个值。1 <= x <= n。

输出格式：
输出一个整数，表示对应的答案

输入输出样例

输入样例#1：
3 4
1 2 3
1 2
2 3
2 1
3 4
输出样例#1：
3
说明

对于20%数据所有数字均为正整数，且小于等于300

对于50%数据所有数字均为正整数，且小于等于3,000

对于100%数据所有数字均为正整数，且小于等于100,000

分析：
子序列可以想到dp，这题显然就是一个限制不同的最长子序列。
我们设$l[i](r[i])$表示这个位置能变成的最小（最大）的值（显然不变也要记录）。
显然这个位置能加入子序列，对于上一个在子序列中的$j$，有
$j<i$
$a[j]<=l[i]$
$r[j]<=a[i]$
就是当前为变成最小也比上一位大，上一位变成最大也没有当前位大。
显然可以k-d tree维护（其实应该叫做二维线段树）。cdq当然也能做，中序遍历cdq就可以转移dp。但是k-d tree太显然了呀。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>

const int maxn=1e5+7;
const int maxd=1e5;

using namespace std;

int n,m,root,cnt,ans,x,y;
int l[maxn],r[maxn],a[maxn],f[maxn];

struct rec{
    int d[2];
};

bool operator ==(rec a,rec b)
{
    return (a.d[0]==b.d[0]) && (a.d[1]==b.d[1]); 
}

struct node{
    int l,r,data;
}t[maxn*100];

void ins(int &p,rec l,rec r,rec d,int k,int op)
{
    if (!p) p=++cnt;
    t[p].data=max(t[p].data,k);
    if (l==r) return;
    int mid=(l.d[op]+r.d[op])/2;
    rec L=l,R=r;
    L.d[op]=mid+1,R.d[op]=mid;
    if (d.d[op]<=mid) ins(t[p].l,l,R,d,k,op^1);
                 else ins(t[p].r,L,r,d,k,op^1);
}

void query(int p,rec l,rec r,rec a,rec b,int op)
{
    if ((l==a) && (r==b)) ans=max(ans,t[p].data);
    if (ans>=t[p].data) return;
    int mid=(l.d[op]+r.d[op])/2;
    rec L=l,R=r;
    L.d[op]=mid+1,R.d[op]=mid;
    if (b.d[op]<=mid) query(t[p].l,l,R,a,b,op^1);
    else if (a.d[op]>mid) query(t[p].r,L,r,a,b,op^1);
    else
    {
        rec A=a,B=b;
        A.d[op]=mid+1,B.d[op]=mid;
        query(t[p].l,l,R,a,B,op^1);
        query(t[p].r,L,r,A,b,op^1);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<=n;i++) l[i]=a[i],r[i]=a[i];
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        l[x]=min(l[x],y);
        r[x]=max(r[x],y);
    }	
    for (int i=1;i<=n;i++)
    {
        ans=0;
        query(root,(rec){{1,1}},(rec){{maxd,maxd}},(rec){{1,1}},(rec){{l[i],a[i]}},0);
        f[i]=ans+1;
        ins(root,(rec){{1,1}},(rec){{maxd,maxd}},(rec){{a[i],r[i]}},f[i],0);
    }
    ans=0;
    for (int i=1;i<=n;i++) ans=max(ans,f[i]);
    printf("%d",ans);
}