当前位置: 代码迷 >> 综合 >> 线段树优化DP [Poi2010]Monotonicity 2
  详细解决方案

线段树优化DP [Poi2010]Monotonicity 2

热度:64   发布时间:2023-12-08 15:12:04.0

问题 B: [Poi2010]Monotonicity 2

时间限制: 1 Sec   内存限制: 256 MB
提交: 57   解决: 40
[ 提交][ 状态][ 讨论版]

题目描述

给出N个正整数a[1..N],再给出K个关系符号(>、<或=)s[1..k]。
选出一个长度为L的子序列(不要求连续),要求这个子序列的第i项和第i+1项的的大小关系为s[(i-1)mod K+1]。
求出L的最大值。

输入

第一行两个正整数,分别表示N和K (N, K <= 500,000)。
第二行给出N个正整数,第i个正整数表示a[i] (a[i] <= 10^6)。
第三行给出K个空格隔开关系符号(>、<或=),第i个表示s[i]。

输出

一个正整数,表示L的最大值。

样例输入

7 3
2 4 3 1 3 5 3
< > =

样例输出

6

提示

选出的子序列为2 4 3 3 5 3,相邻大小关系分别是< > = < >。

         f[i]表示到i位最大子序列长度。转移由之前某一个满足的条件转移(由f[j]的大小判断匹配到第几个条件)
        当然上面的都只是暴力。。
        转移可分别由<,>,=转移得到,只来自之前,所以可以用线段树维护区间最大值。以此节点的权值作为线段树的下标,f[i]为值,并根据f[i]的大小考虑该推进<,>的线段树中(共两棵,得分别维护)查询时只要查询比a[i](大/小)的区间就行了。
        关于等于号,不用再开线段树,只要记录上一个相等且该取=的大小就行了。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 500000
using namespace std;
int n,m,a[N+5],f[N+5],b[N+5];
char s[5];
int read()
{int sum=0,f=1;char x=getchar();while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}while(x>='0'&&x<='9')sum=sum*10+x-'0',x=getchar();return sum*f;
}
int main()
{//freopen("mot.in","r",stdin);//freopen("mot.out","w",stdout);n=read();m=read();for(int i=1;i<=n;i++)a[i]=read();for(int i=1;i<=m;i++){scanf("%s",s);if(s[0]=='<')b[i]=0;if(s[0]=='>')b[i]=1;if(s[0]=='=')b[i]=2;}f[1]=1;for(int i=1;i<=n;i++)for(int j=1;j<i;j++){int k=f[j]%m;if(b[k]==0&&a[j]<a[i])f[i]=max(f[i],f[j]+1);if(b[k]==1&&a[j]>a[i])f[i]=max(f[i],f[j]+1);if(b[k]==2&&a[j]==a[i])f[i]=max(f[i],f[j]+1);}//cout<<sum;  
}