CF868F Yet Another Minimization Problem

题目描述

You are given an array of nn integers a_{1}...\ a_{n}a1​... an​ . The cost of a subsegment is the number of unordered pairs of distinct indices within the subsegment that contain equal elements. Split the given array into kk non-intersecting non-empty subsegments so that the sum of their costs is minimum possible. Each element should be present in exactly one subsegment.

输入格式

The first line contains two integers nn and kk ( 2<=n<=10^{5}2<=n<=105 , 2<=k<=min\ (n,20))2<=k<=min (n,20)) — the length of the array and the number of segments you need to split the array into.

The next line contains nn integers a_{1},a_{2},...,a_{n}a1​,a2​,...,an​ ( 1<=a_{i}<=n1<=ai​<=n ) — the elements of the array.

输出格式

Print single integer: the minimum possible total cost of resulting subsegments.

题意翻译

题目描述：给定一个序列 aa，要把它分成 kk 个子段。每个子段的费用是其中相同元素的对数。求所有子段的费用之和的最小值。

输入格式：第一行输入 nn（序列长度）和 kk（需分子段数）。接下来一行有 nn 个数，第 ii 个数表示序列的第 ii 个元素 a_iai​。

输出格式：输出一个数，费用和的最小值。

2 \leq n \leq 10^52≤n≤105，2 \leq k \leq min(n,20)2≤k≤min(n,20)，1 \leq a_i \leq n1≤ai​≤n 。

输入输出样例

输入 #1复制

7 3
1 1 3 3 3 2 1

输出 #1复制

1

输入 #2复制

10 2
1 2 1 2 1 2 1 2 1 2

输出 #2复制

8

输入 #3复制

13 3
1 2 2 2 1 2 1 1 1 2 2 1 1

输出 #3复制

9

说明/提示

In the first example it's optimal to split the sequence into the following three subsegments: [1][1] , [1,3][1,3] , [3,3,2,1][3,3,2,1] . The costs are 00 , 00 and 11 , thus the answer is 11 .

In the second example it's optimal to split the sequence in two equal halves. The cost for each half is 44 .

In the third example it's optimal to split the sequence in the following way: [1,2,2,2,1][1,2,2,2,1] , [2,1,1,1,2][2,1,1,1,2] , [2,1,1][2,1,1] . The costs are 44 , 44 , 11 .

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int c[100005],tr,tl,f[30][100005],a[100005],s,n,k;
int get(int l,int r)
{
    while (tl<l) 
    {
        c[a[tl]]--;
        s-=c[a[tl]];
        tl++;
    }
    while (tl>l)
    {
        tl--;
        s+=c[a[tl]];
        c[a[tl]]++;
    }
    while (tr<r)
    {
        tr++;
        s+=c[a[tr]];
        c[a[tr]]++;
    }
    while (tr>r)
    {    
        c[a[tr]]--;
        s-=c[a[tr]];
        tr--;
    } 
    return s;
}
void solve(int x,int l,int r,int L,int R)
{int i,Mid;
    if (l>r) return;
    int mid=(l+r)/2;
    Mid=mid;
    for (i=L;i<=min(mid-1,R);i++)
    {
        int t=f[x-1][i]+get(i+1,mid);
        if (f[x][mid]>=t) 
        {
            f[x][mid]=t;
            Mid=i;
        }
    }
    solve(x,l,mid-1,L,Mid);
    solve(x,mid+1,r,Mid,R);
}
int main()
{int i,j;
    scanf("%d%d",&n,&k);
    for (i=1;i<=n;i++)
    scanf("%d",&a[i]);
    memset(f,127/2,sizeof(f));
    tl=1;tr=0;
    for (i=1;i<=n;i++)
    {
        f[1][i]=get(1,i);
    } 
    for (i=2;i<=k;i++)
    {
        solve(i,i,n,i-1,n);
    }
    printf("%d",f[k][n]);
}