题目链接:http://codeforces.com/problemset/problem/834/D
题意:把n个数分成k段,每段的价值为这一段数里不同数字的个数,问价值和最大为多少。
思路:我们用dp[i][j]表示前j个数分成i段价值和的最大值,那么可以得到转移方程式:
dp[i][j]=max(dp[i-1][k]+qw(k+1,j))(0=<k<j)
如果直接进行转移的话肯定会T掉,所以需要用线段树加速转移,我们可以先把 dp i- 1层的全部扔进线段树,然后不断更新区间不同的数,相当于把这个转移方程全部扔进线段树,然后再询问一个区间最大值,更新新的dp值,线段树每个节点存的值是
dp[i-1][k]+qw(k+1,j),而dp[i-1][k]是一个定值,所以从dp[i-1][k]到dp[i][k]所需要考虑的是qw(k+1,j)的变化,那么考虑第j个数,它会对区间【b[j],j】区间内的树上的位置有所影响。(b[j]表示a[j]上一次出现的位置+1)
那么我们遍历到第j个位子的时候,将树上区间【b[j],j】的值都加一。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define rc (d<<1|1)
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int n,k,a[35008],dp[58][35008],b[35008],c[35008];
struct as{
int l,r,mx,lz;}tr[35008<<2];
void push(int d){tr[d].mx=max(tr[lc].mx,tr[rc].mx);}
void build(int d,int l,int r,int x)
{
tr[d].l=l,tr[d].r=r,tr[d].lz=0;
if(l==r) {tr[d].mx=dp[x][l-1];return;}
int mid=(l+r)>>1;
build(lc,l,mid,x);
build(rc,mid+1,r,x);
push(d);
}
void as(int d)
{
if(tr[d].lz)
{
tr[lc].lz+=tr[d].lz,tr[rc].lz+=tr[d].lz;
tr[lc].mx+=tr[d].lz,tr[rc].mx+=tr[d].lz;
tr[d].lz=0;
}
}
void add(int d,int l,int r)
{
if(l==tr[d].l&&r==tr[d].r) {tr[d].lz++,tr[d].mx++;return;}
as(d);
int mid=(tr[d].l+tr[d].r)>>1;
if(mid>=r) add(lc,l,r);
else if(l>mid) add(rc,l,r);
else add(lc,l,mid),add(rc,mid+1,r);
push(d);
}
int query(int d,int l,int r)
{
if(tr[d].l==l&&tr[d].r==r) return tr[d].mx;
as(d);
int mid=(tr[d].l+tr[d].r)>>1;
if(mid>=r) return query(lc,l,r);
else if(l>mid) return query(rc,l,r);
else return max(query(lc,l,mid),query(rc,mid+1,r));
}
int main()
{
cin.tie(0);
cout.tie(0);
cin>>n>>k;
FOR(i,1,n) si(a[i]);
FOR(i,1,n) b[i]=c[a[i]]+1,c[a[i]]=i;
FOR(i,1,k)
{
build(1,1,n,i-1);
FOR(j,1,n)
{
add(1,b[j],j);
dp[i][j]=query(1,1,j);
}
}
cout<<dp[k][n]<<endl;
return 0;
}