正題

題目鏈接:https://www.luogu.com.cn/problem/P4983

---

題目大意

給出長度為$n$的序列$x$，記平均數為$\bar{x}$，要求將序列分成$m$段。
每一段$[l,r]$的值為
$$\frac{(({\sum }_{i=l}^r{x}_i\times \bar{x})+\bar{x}{)}^2}{{\bar{x}}^2}$$

求所有段的值和最小

$1\le m\le n\le 10^5,1\le x_i\le 1000$

---

解題思路

直接除以${\bar{x}}^2$就是最小化$({\sum }_{i=l}^r{x}_i+1{)}^2$的和。

然后這個問題是下凸函數，設$f(i)$表示恰好分成$i$段，那么顯然段數越多答案越小而且每次減少的越少。

所以我們可以用$wqs$二分給每次分一個段加上一個權值$val$。

那么現在的轉移就是
$$F_i=min\{F_j+(s_i?s_j+1{)}^2+val\}(j<i)$$

這是經典的斜率優化不過多贅述。

時間復雜度$O(n\log W)$（$W$表示二分值域）

---

code

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=1e5+10; ll n,m,s[N],f[N],g[N],x[N],y[N],q[N]; ll count(ll l,ll r) {return f[l]+(s[r]-s[l]+1)*(s[r]-s[l]+1);} ll xj(ll p,ll q,ll z) {return (x[p]-x[z])*(y[q]-y[z])-(x[q]-x[z])*(y[p]-y[z]);} ll check(ll val){int head=1,tail=0;q[++tail]=0;for(ll i=1;i<=n;i++){while(head<tail&&2ll*s[i]*(x[q[head+1]]-x[q[head]])>(y[q[head+1]]-y[q[head]]))head++;f[i]=count(q[head],i)+val;g[i]=g[q[head]]+1;y[i]=f[i]+s[i]*s[i]-2*s[i];x[i]=s[i];while(head<tail&&xj(i,q[tail],q[tail-1])>=0)tail--;q[++tail]=i;}return g[n]; } signed main() {scanf("%lld%lld",&n,&m);for(ll i=1;i<=n;i++)scanf("%lld",&s[i]),s[i]+=s[i-1];ll l=0,r=1e18;while(l<=r){ll mid=(l+r)>>1;if(check(mid)<=m)r=mid-1;else l=mid+1;}check(l);printf("%lld\n",f[n]-l*m);return 0; }

總結

以上是生活随笔為你收集整理的P4983-忘情【wqs二分,斜率优化】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。