Codeforces 834D - The Bakery 【DP+线段树】

题目链接：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;
}