BZOJ3437: 小P的牧場(斜率優化DP)
阿新 • • 發佈:2018-11-21
3437: 小P的牧場
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 1914 Solved: 1046
[Submit][Status][Discuss]
Description
小P在MC裡有n個牧場,自西向東呈一字形排列(自西向東用1…n編號),於是他就煩惱了:為了控制這n個牧場,他需要在某些牧場上面建立控制站,每個牧場上只能建立一個控制站,每個控制站控制的牧場是它所在的牧場一直到它西邊第一個控制站的所有牧場(它西邊第一個控制站所在的牧場不被控制)(如果它西邊不存在控制站,那麼它控制西邊所有的牧場),每個牧場被控制都需要一定的花費(畢竟在控制站到牧場間修建道路是需要資源的嘛~),而且該花費等於它到控制它的控制站之間的牧場數目(不包括自身,但包括控制站所在牧場)乘上該牧場的放養量,在第i個牧場建立控制站的花費是ai,每個牧場i的放養量是bi,理所當然,小P需要總花費最小,但是小P的智商有點不夠用了,所以這個最小總花費就由你來算出啦。
Input
第一行一個整數 n 表示牧場數目
第二行包括n個整數,第i個整數表示ai
第三行包括n個整數,第i個整數表示bi
Output
只有一行,包括一個整數,表示最小花費
Sample Input
42424
3142
Sample Output
9樣例解釋
選取牧場1,3,4建立控制站,最小費用為2+(2+1*1)+4=9。
1<=n<=1000000, 0 < a i ,bi < = 10000
HINT
Source
沒什麼好說的了感覺。
#include<bits/stdc++.h> #define ll long long #define rep(i,a,b) for(int i=a;i<=b;i++) using namespace std; const int maxn=1000010; int a[maxn],b[maxn],q[maxn],top,head,N; ll X[maxn],sum[maxn],dp[maxn]; ll DY(int i){ return dp[i]+X[i];} ll getans(int i,int j){ return dp[j]+a[i]+(sum[i]-sum[j])*i-(X[i]-X[j]); }int main() { scanf("%d",&N); rep(i,1,N) scanf("%d",&a[i]); rep(i,1,N) scanf("%d",&b[i]); rep(i,1,N) { sum[i]=sum[i-1]+b[i]; X[i]=X[i-1]+1ll*b[i]*i; } rep(i,1,N){ while(head<top&&getans(i,q[head])>=getans(i,q[head+1])) head++; dp[i]=getans(i,q[head]); while(top>0&&(DY(i)-DY(q[top]))*(sum[i]-sum[q[top-1]])<(DY(i)-DY(q[top-1]))*(sum[i]-sum[q[top]])) top--; q[++top]=i; } printf("%lld\n",dp[N]); return 0; }
二分的版本,在斜率不送單調的時候用:
#include<bits/stdc++.h> #define ll long long #define rep(i,a,b) for(int i=a;i<=b;i++) using namespace std; const int maxn=1000010; ll a[maxn],b[maxn],q[maxn],top,N; ll X[maxn],sum[maxn],dp[maxn]; ll DY(int i){ return dp[i]+X[i];} ll getans(int i,int j){ return dp[j]+a[i]+(sum[i]-sum[j])*i-(X[i]-X[j]); } int main() { scanf("%lld",&N); rep(i,1,N) scanf("%lld",&a[i]); rep(i,1,N) scanf("%lld",&b[i]); rep(i,1,N) { sum[i]=sum[i-1]+b[i]; X[i]=X[i-1]+b[i]*i; } rep(i,1,N){ int L=0,R=top,Mid,pos=top; while(L<=R){ Mid=(L+R)>>1; if(getans(i,q[Mid])<=getans(i,q[Mid+1])) pos=Mid,R=Mid-1; else L=Mid+1; } dp[i]=getans(i,q[pos]); while(top>0&&(DY(i)-DY(q[top]))*(sum[i]-sum[q[top-1]])<(DY(i)-DY(q[top-1]))*(sum[i]-sum[q[top]])) top--; q[++top]=i; } printf("%lld\n",dp[N]); return 0; }View Code