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

 

posted @ 2021-08-05 22:11  智人心  阅读(71)  评论(0)    收藏  举报