一、题目
二、解法
暴力的
d
p
dp
dp比较好想,设
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]为前
i
i
i个里面划分
j
j
j块,转移如下:
d
p
[
i
]
[
j
]
=
d
p
[
k
]
[
j
−
1
]
+
c
dp[i][j]=dp[k][j-1]+c
dp[i][j]=dp[k][j−1]+c其中
c
c
c为当前块的价值,从前往后扫就可以维护,时间复杂度
O
(
n
2
k
)
O(n^2k)
O(n2k)。
观察上面的方程,好像可以用线段树优化,但是当前块的贡献怎么算呢?我们在 i i i右移的过程中维护这个贡献,预处理出 p [ i ] p[i] p[i]为和 i i i位置权值一样的最近的位置,把 [ p [ i ] , i − 1 ] [p[i],i-1] [p[i],i−1] 这个区间 + 1 +1 +1,表示选取这些转移点会多 1 1 1的贡献。
时间复杂度 O ( n k log n ) O(nk\log n) O(nklogn),贴个代码 q w q qwq qwq。
#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 35005;
int read()
{
int x=0,flag=1;char c;
while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*flag;
}
int n,k,p[M],dp[M],mx[4*M],la[4*M];
struct node
{
int x,p;
node(int X=0,int P=0) : x(X) , p(P) {}
bool operator < (const node &B) const
{
if(x==B.x) return p<B.p;
return x<B.x;
}
}a[M];
void build(int i,int l,int r)
{
la[i]=0;
if(l==r)
{
mx[i]=dp[l];
return ;
}
int mid=(l+r)>>1;
build(i<<1,l,mid);
build(i<<1|1,mid+1,r);
mx[i]=max(mx[i<<1],mx[i<<1|1]);
}
void down(int i)
{
if(!la[i]) return ;
int ls=i<<1,rs=i<<1|1;
la[ls]+=la[i];
la[rs]+=la[i];
mx[ls]+=la[i];
mx[rs]+=la[i];
la[i]=0;
}
void upd(int i,int l,int r,int L,int R)
{
if(l>R || L>r) return ;
if(L<=l && r<=R)
{
la[i]++;mx[i]++;
return ;
}
down(i);
int mid=(l+r)>>1;
upd(i<<1,l,mid,L,R);
upd(i<<1|1,mid+1,r,L,R);
mx[i]=max(mx[i<<1],mx[i<<1|1]);
}
int ask(int i,int l,int r,int L,int R)
{
if(l>R || L>r) return 0;
if(L<=l && r<=R) return mx[i];
down(i);
int mid=(l+r)>>1;
return max(ask(i<<1,l,mid,L,R),ask(i<<1|1,mid+1,r,L,R));
}
int main()
{
n=read();k=read();
for(int i=1;i<=n;i++)
a[i]=node(read(),i);
sort(a+1,a+1+n);
for(int i=1,j=0;i<=n;i=j)
{
j=i+1;
for(;a[i].x==a[j].x;j++)
p[a[j].p]=a[j-1].p;
}
for(int j=1;j<=k;j++)
{
build(1,0,n);
for(int i=1;i<=n;i++)
{
upd(1,0,n,p[i],i-1);
dp[i]=ask(1,0,n,0,i-1);
}
}
printf("%d\n",dp[n]);
}