Codeforces 833B - The Bakery 【DP+线段树】

B. The Bakery

time limit per test 2.5seconds
memory limit per test     256megabytes
 

Some time ago Slastyona the Sweetmaid decided to open her ownbakery! She bought required ingredients and a wonder-oven which can bakeseveral types of cakes, and opened the bakery.

Soon the expenses started to overcome the income, so Slastyonadecided to study the sweets market. She learned it's profitable to pack cakesin boxes, and that the moredistinct cake types a box contains (let's denote this number as the value of the box), the higherprice it has.

She needs to change the production technology! The problem isthat the oven chooses the cake types on its own and Slastyona can't affect it.However, she knows the types and order ofn cakes the oven is going to bake today.Slastyona has to pack exactly k boxes with cakes today, and she has to put in each box several(at least one) cakes the oven produced oneright after another (in other words, shehas to put in a box a continuous segment of cakes).

Slastyona wants to maximize the total value of all boxes withcakes. Help her determine this maximum possible total value.

Input

The first line contains two integersn and k (1 ≤ n ≤ 35000,1 ≤ k ≤ min(n, 50)) –the number of cakes and the number of boxes, respectively.

The second line containsn integers a1, a2, ..., an (1 ≤ ai ≤ n) – the types of cakes in the order the oven bakes them.

Output

Print the only integer – the maximum total value of all boxes with cakes.

【题意】

把n个数分成k段，每段的价值为这一段数里不同数字的个数，问价值和最大为多少。

【思路】

显然需要用到动态规划。我们用dp[i][j]表示前j个数分成i段价值和的最大值。容易得到状态转移方程为

dp[i][j]=max{dp[i-1][x]+剩下数的贡献} {1<=x< j}

过程用线段树维护一下区间最大值即可。
 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<vector>

using namespace std;

const int maxn = 35005;

int n,k;
int dp[55][maxn];
int tree[maxn<<2],lazy[maxn<<2];

int a[maxn],pre[maxn],pos[maxn];

void init()
{
    memset(tree,0,sizeof(tree));
    memset(pos,0,sizeof(pos));
    memset(pre,0,sizeof(pre));
    memset(dp,0,sizeof(dp));
}

void pushup(int rt)
{
    tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}

void pushdown(int rt)
{
    tree[rt<<1]+=lazy[rt];
    tree[rt<<1|1]+=lazy[rt];
    lazy[rt<<1]+=lazy[rt];
    lazy[rt<<1|1]+=lazy[rt];
    lazy[rt]=0;
}

void build(int pos,int l,int r,int rt)
{
    lazy[rt]=0;
    if(l==r)
    {
        tree[rt]=dp[pos][l-1];
        return ;
    }
    int m=(l+r)/2;
    build(pos,l,m,rt<<1);
    build(pos,m+1,r,rt<<1|1);
    pushup(rt);
}

void update(int L,int R,int l,int r,int rt)
{
    if(L==l && R==r)
    {
        tree[rt]++;
        lazy[rt]++;
        return ;
    }
    pushdown(rt);
    int m=(l+r)/2;

    if(L>m)
        update(L,R,m+1,r,rt<<1|1);
    else if(R<=m)
        update(L,R,l,m,rt<<1);
    else
    {
        update(L,m,l,m,rt<<1);
        update(m+1,R,m+1,r,rt<<1|1);
    }
    pushup(rt);
}

int query(int L,int R,int l,int r,int rt)
{
    if(L==l && R==r)
    {
        return tree[rt];
    }
    pushdown(rt);
    int m=(l+r)/2;
    int ans=0;
    if(L>m)
        ans=max(ans,query(L,R,m+1,r,rt<<1|1));
    else if(R<=m)
        ans=max(ans,query(L,R,l,m,rt<<1));
    else
    {
        ans=max(ans,max(query(L,m,l,m,rt<<1),query(m+1,R,m+1,r,rt<<1|1)));
    }

    return ans;
}

int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pre[i]=pos[a[i]]+1;
            pos[a[i]]=i;
        }

        for(int i=1;i<=k;i++)
        {
            build(i-1,1,n,1);
            for(int j=1;j<=n;j++)
            {
                update(pre[j],j,1,n,1);
                dp[i][j]=query(1,j,1,n,1);
            }
        }
        printf("%d\n",dp[k][n]);
    }
}