2020 CCPC Wannafly Winter Camp Day3 F. 社团管理 (DP && 分治)

题意：

有$n$个数字$a_{1,2...n}$,要分成k段，每段的价值为段内满足$[a_i==a_j]and[i==j]$的$(i,j)$对数。
$(n,a_i\le 1e5,k\le min(n,20))$

解题思路：

有一个很显然的$O(n^{2}k)$的做法,$dp(i,j)$表示把$a_{1,2,..j}$分成$i$段的最小价值，转移为
$dp(i,j)=min(dp(i-1,j^{\prime })+cost(j^{\prime },j))[j^{\prime }<j]$
这里$cost(j^{\prime },j)$表示$(j^{\prime },j]$这一段的价值。
考虑优化，往单调性方面去考虑：每个转移点的位置是否满足单调性？设$p_j$为$dp(i,j)$的最左端的最优转移点，对于$(j^{\prime }<j)$是否总有$p_{j^{\prime }}\le p_j$？
答案是Yes。现在用反证法证明：
证明思路来自：https://codeforces.com/blog/entry/55046
假设存在$j^{\prime }<j$，且$dp(i-1,x)+cost(x,j)<dp(i-1,p_{j^{\prime }})+cost(p_{j^{\prime }},j)$满足$x<p_{j^{\prime }}$。
画图出来是这样：

那么显然的有：
$cost(x,j)-cost(p_{j^{\prime }},j)>cost(x,j^{\prime })-cost(p_{j^{\prime }},j^{\prime })$
由假设的条件我们得到
$dp(i-1,p_{j^{\prime }})-dp(i-1,x)>cost(x,j)-cost(p_{j^{\prime }},j)>cost(x,j^{\prime })-cost(p_{j^{\prime }},j^{\prime })$
移项得到$dp(i-1,p_{j^{\prime }})+cost(p_{j^{\prime }},j^{\prime })>dp(i-1,x)+cost(x,j^{\prime })$
这与$p_{j^{\prime }}$是$j^{\prime }$的最左端最优转移点矛盾
证毕。
那么知道了它满足单调性之后，就可以采用类似整体二分的分治去转移了，因为每一层的搜索区间总长度是$O(n)$的，一共有$log$层，所以总复杂度$O(knlogn)$
代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5 + 50;
int cnt[maxn], a[maxn];
ll ans;
void add(int x){
    ans += cnt[x]; cnt[x]++;
}
void del(int x){
    ans -= (cnt[x]-1); cnt[x]--;
}
int lp , rp;
void go(int l, int r){
    while(lp > l) add(a[--lp]);
    while(rp < r) add(a[++rp]);
    while(lp < l) del(a[lp++]);
    while(rp > r) del(a[rp--]);
}
ll dp[21][maxn];
int n, k, t;
void sol(int l, int r, int L, int R, ll *pre, ll *cur){
    if(l > r) return;
    int mid = (l+r)>>1, p;
    for(int i = L; i <= min(R, mid-1); ++i){
        go(i+1, mid);
        if(cur[mid] > pre[i] + ans) cur[mid] = pre[i] + ans, p = i;
    }
    sol(l, mid-1, L, p, pre, cur); sol(mid+1, r, p, R, pre, cur);
}
int main()
{
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
    memset(dp, 0x3f, sizeof dp); dp[0][0] = 0;
    lp = rp = 1; add(a[1]);
    for(t = 1; t <= k; ++t){
        dp[t][0] = 0;
        sol(1, n, 0, n-1, dp[t-1], dp[t]);
    }
    cout<<dp[k][n]<<endl;
}
/*
20 3
1 3 2 0 1 0 2 2 2 0 1 1 1 3 1 3 3 2 3 0
*/