poj 1701(公式推導,注意理解題意,上樓是從當前停的樓層到目標樓層,而不是從當前樓層上到目標樓層坐電梯)
阿新 • • 發佈:2021-08-05
#include<iostream> #include<cstdio> using namespace std; #define ll long long ll m,a,b,k[10005]; ll get_ans(ll n){ ll i,tmp = 0; for(i=1;i<n;i++) tmp += k[i]*((n-i)*b+(n-i)*(n-i-1)/2); for(i=n+1;i<=m;i++) tmp += k[i]*((i-n)*a+(i-n)*(i-n-1)/2); return tmp; }int main(){ ll smin,si,sum,t,i,j,dp[10005],ans[10005]; scanf("%lld",&t); while(t--){ scanf("%lld%lld%lld",&m,&a,&b); for(i=1;i<=m;i++) scanf("%lld",&k[i]); ans[1] = get_ans(1); ans[2] = get_ans(2); for(i=1,sum=0;i<=m;i++) sum+= k[i]; dp[1] = ans[2]-ans[1]; for(i=2;i<m;i++){ dp[i] = k[i]*(a+b-1)+sum+dp[i-1]; } for(i=3;i<=m;i++) ans[i] = ans[i-1]+dp[i-1]; smin = (unsigned ll)-1>>1; si = -1; for(i=1;i<=m;i++) if(smin>ans[i]){ smin= ans[i]; si = i; } printf("%lld\n",si); } return 0; }