斜率优化--bzoj4709

最新推荐文章于 2020-02-21 23:30:46 发布

QAQQQQQQQQQQQ

最新推荐文章于 2020-02-21 23:30:46 发布

阅读量162

点赞数

分类专栏：斜率优化DP 文章标签：斜率优化

本文链接：https://blog.csdn.net/sizeof_you/article/details/81160561

版权

斜率优化DP 专栏收录该内容

2 篇文章 1 订阅

订阅专栏

根据题目可列出转移方程：
f[i]=f[j-1]+s[i]*(sum[i]-sum[j]+1)^2 (s[i]==s[j])
其中sum表示的是从1开始a[i]这个值出现了几次，f[i]就是从1开始到i的答案

通过拆开移项，我们可以得到一个斜率优化的式子：
f[j-1]+s[j](sum[j]-1)^2=2 s[i] * sum[i] (sum[j]-1)+f[i]-sum[i] s[i]

但是我们发现斜率不是单调的
所以要二分一下
（我大概是傻了调了一晚上）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define maxn 100005
#define LL long long
using namespace std;
int s[maxn],n,l,r,ret;
LL cnt[10010],sum[maxn],f[maxn];
vector<int> a[10010];

inline int rd(){
  int x=0,f=1;char c=' ';
  while(c<'0' || c>'9') {if(c=='-') f=-1;c=getchar();}
  while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
  return x*f;
}

inline LL squ(LL x) {return x*x;}
#define y(i) (f[i-1]+squ(sum[i]-1)*s[i])
#define x(i) 2*(sum[i]-1)

int main(){
  n=rd();
  for(int i=1;i<=n;i++){
    s[i]=rd();
    cnt[s[i]]++;//预处理
    sum[i]=cnt[s[i]];
  }
  for(int i=1;i<=n;i++){
    int t;
    while((t=a[s[i]].size()-1)>0 && (x(i) - x(a[s[i]][t])) * (y(a[s[i]][t - 1]) - y(a[s[i]][t])) - (y(i) - y(a[s[i]][t])) * (x(a[s[i]][t - 1]) - x(a[s[i]][t])) > 0) a[s[i]].pop_back();
    a[s[i]].push_back(i);
    l=1,r=a[s[i]].size()-1,ret=0;
    while(l<=r){
      int mid=(l+r)>>1;
      if(f[a[s[i]][mid]-1]+s[i]*squ(sum[i]-sum[a[s[i]][mid]]+1)>f[a[s[i]][mid-1]-1]+s[i]*squ(sum[i]-sum[a[s[i]][mid-1]]+1)) ret=mid,l=mid+1;
      else r=mid-1;
    }
    int x=a[s[i]][ret];
    f[i]=f[x-1]+s[i]*squ(sum[i]-sum[x]+1);
  }
  printf("%lld\n",f[n]);
  return 0;
}