1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
| #include <bits/stdc++.h>
using namespace std;
const double eps = 1e-10; const double pi = 3.1415926535897932384626433832795; const double eln = 2.718281828459045235360287471352;
#define f(i, a, b) for (int i = a; i <= b; i++) #define scan(x) scanf("%d", &x) #define mp make_pair #define pb push_back #define lowbit(x) (x&(-x))
#define fi first #define se second #define SZ(x) int((x).size()) #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define summ(a) (accumulate(all(a), 0ll))
typedef unsigned long long ull; typedef pair<int,int> pii; typedef vector<int> vi;
using ll=long long;
const int maxn=2e3+9; int n,m,rt[maxn],cnt,b[maxn],a[maxn],tot; struct Node{ int l,r,val; }tr[maxn*20];
inline void build(int &k,int l,int r){ k=++cnt; if(l==r)return; int mid=(l+r)>>1; build(tr[k].l,l,mid),build(tr[k].r,mid+1,r); }
inline void change(int &k,int kk,int l,int r,int val){ k=++cnt,tr[k]=tr[kk],tr[k].val+=1; if(l==r)return ; int mid=(l+r)>>1; if(val<=mid)change(tr[k].l,tr[kk].l,l,mid,val); else change(tr[k].r,tr[kk].r,mid+1,r,val); }
inline int query(int nl,int nr,int l,int r,int k){ int x=tr[tr[nr].l].val-tr[tr[nl].l].val; if(l==r)return b[l]; int mid=(l+r)>>1; if(x>=k)return query(tr[nl].l,tr[nr].l,l,mid,k); return query(tr[nl].r,tr[nr].r,mid+1,r,k-x); }
ll f[109][109][109]; ll dfs(ll l,ll r,ll cs){ if(f[l][r][cs])return f[l][r][cs]; if(cs==1){ int len=r-l+1; return f[l][r][cs]=query(rt[l-1],rt[r],1,tot,len-len/20); } ll res=0x3f3f3f3f; for(int i=l;i<n;++i){ res=min(res,dfs(l,i,1)+dfs(i+1,n,cs-1)); } return f[l][r][cs]=res; }
void solve(){ memset(tr,0,sizeof(tr)); memset(f,0,sizeof(f)); cin>>n>>m; f(i,1,n)cin>>a[i],b[i]=a[i]; sort(b+1,b+1+n); tot=unique(b+1,b+n+1)-(b+1); build(rt[0],1,tot); f(i,1,n){ int now=lower_bound(b+1,b+1+tot,a[i])-b; change(rt[i],rt[i-1],1,tot,now); } cout<<dfs(1,n,m)<<"\n"; } ll tt; int main() { ios::sync_with_stdio(false); cin.tie(0); cin>>tt; f(sb,1,tt)solve(); return 0; }
|