2021湘潭邀请赛-正式赛-I-区间划分dp,线段树优化贡献转移

2023大厂真题提交网址(含题解):

www.CodeFun2000.com（http://101.43.147.120/）

最近我们一直在将收集到的机试真题制作数据并搬运到自己的OJ上，供大家免费练习,体会真题难度。现在OJ已录入50+道2023年最新大厂真题，同时在不断的更新。同时，可以关注"塔子哥学算法"公众号获得每道题的题解。

类似题目:CFRound7221-E

前言:

本题给出了一种新的思路来用线段树优化dp,即用线段树优化花费函数转移.

题目大意:

给你一个$n$序列。让你将其划分成$k$段。每一段的贡献为:该段不同数字个数,问最小化贡献和。
$n\le 1e5,k\le 10$

题目思路:

暴力dp.
$dp(i,j)$代表前$i$个数分成$j$个段的最小贡献和。有$O(n^{2}k)$的做法。

d p ( i , j ) = m i n { d p ( k , j − 1 ) + d i f ( k + 1 , i ) } dp(i,j)=min\{dp(k,j-1)+dif(k+1,i)\} dp(i,j)=min{dp(k,j−1)+dif(k+1,i)}. 难点就在于$dif$函数的求解.

提出一种高效维护$dif$函数的算法:

考虑新增$a[i]$对$dif$函数的影响,即从$dif(k+1,i-1),k\in [j-1,i-1]$转移到$dif(k+1,i)$：

令$last[i]$代表$a[i]$的上一次出现的位置.

那么当$k\in [j-1,last[i]-1)$时,$dif(k+1,i)=dif(k+1,i-1)$.
当$k\in [last[i]-1,i-1]$时,$dif(k+1,i)=dif(k+1,i-1)+1$.

根据贡献的区间性,我们可以考虑用线段树来维护$dif$函数.

当然，我们就可以直接维护$dp(k,j-1)+dif(k+1,i)$。对每个点，先区间修改实现转移，再区间查询最小值即可。

AC代码:(这其实是cf的，懒得改了，基本一样).

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
#define tl (t<<1)
#define tr (t<<1|1)
#define mid ((r + l) >> 1)
const int maxn = 35000 + 5;
const int maxm = 105;
const int mod = 1e9 + 7;
const int inf = 1e9;
int mi[maxn << 2] , lazy[maxn << 2],a[maxn];
void pushup(int t){
    mi[t] = min(mi[tl] , mi[tr]);
}
void D(int t , int c){
    mi[t] += c;
    lazy[t] += c;
}
void pushdown(int t)
{
    if(lazy[t]){
        D(tl,lazy[t]);
        D(tr,lazy[t]);
        lazy[t] = 0;
    }
}
void Build(int l,int r,int t)
{
    lazy[t] = 0;
    if(l == r){
        mi[t] = 0;
        return ;
    }
    Build(l , mid , tl);
    Build(mid + 1 , r , tr);
    pushup(t);
}
void Update(int L,int R,int l,int r,int t,int c)
{
    if (L > R) return ;
    if(L <= l && r <= R)
    {
        mi[t] += c;
        lazy[t] += c;
        return ;
    }
    pushdown(t);
    if(L <= mid)
        Update(L,R,l,mid,tl,c);
    if(R > mid)
        Update(L,R,mid+1,r,tr,c);
    pushup(t);
}
int Query(int L,int R,int l,int r,int t)
{
    if(L <= l && r <= R){
        return mi[t];
    }
    pushdown(t);
    int ans = inf;
    if(L <= mid)
        ans = min(ans , Query(L,R,l,mid,tl));
    if(R > mid)
        ans = min(ans , Query(L,R,mid+1,r,tr))  ;
    return ans;
}
int dp[maxn][maxm];
int last[maxn] , pos[maxn];
int main()
{
    ios::sync_with_stdio(false);
    int n , m; cin >> n >> m;
    for (int i = 1 ; i <= n ; i++) {
        cin >> a[i];
        pos[i] = last[a[i]];
        last[a[i]] = i;
    }
    for (int i = 1 ; i <= n ; i++) {
        dp[i][1] = dp[i - 1][1] + (pos[i] == 0 ? 0 : i - pos[i]);
    }
    for (int j = 2 ; j <= m ; j++){
        Build(1 , n , 1); // 清空线段树
        // 将上一个阶段的dp值插入线段树
        for (int i = 1 ; i <= n ; i++) Update(i , i , 1 , n , 1 , dp[i][j - 1]);
        // 针对第i位作dp
        for (int i = 1 ; i < j ; i++) dp[i][j] = inf;
        for (int i = j ; i <= n ; i++){
            if (pos[i]) Update(1 , pos[i] - 1 , 1 , n , 1 , i - pos[i]);
            dp[i][j] = Query(1 , i - 1 , 1 , n , 1);
        }
    }
    cout << dp[n][m] << endl;
    return 0;
}