1. 程式人生 > 其它 >poj 1701(公式推導,注意理解題意,上樓是從當前停的樓層到目標樓層,而不是從當前樓層上到目標樓層坐電梯)

poj 1701(公式推導,注意理解題意,上樓是從當前停的樓層到目標樓層,而不是從當前樓層上到目標樓層坐電梯)

#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; }