poj 1701 Dissatisfying Lift

//自己推公式推了一个多小时，代码写了10分钟。。

//设停在t层的不满意度为ans[t]，推公式可以发现

//ans[t+1]-ans[t]=b*(k1+k2+......+kt)-a*(kt+1 +kt+2....+kn)+kt-1+2*kt-2+......+(t-1)*k1 -kt+2-2*kt+3-......-(n-t-1)kn

// 令dp[t]=ans[t+1]-ans[t]，可以发现dp[t+1]-dp[t]=(a+b-1)*kt+1+ (k1+k2+......kn).

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int k[10005], n, a, b, sk;
long long ans[10005], dp[10005];
long long get_ans(int t){
    long long tmp=0;
    int i;
    for(i=1; i<=t; i++)
        tmp+=(b*(t-i)+(t-i)*(t-i-1)/2)*k[i];
    for(i=t; i<=n; i++)
        tmp+=(a*(i-t)+(i-t)*(i-t-1)/2)*k[i];
    return tmp;
}

int main(){
    //freopen("1.txt", "r", stdin);
    int T, i, t;
    scanf("%d", &T);
    while(T--){
        sk=0;
        scanf("%d%d%d", &n, &a, &b);
        for(i=1; i<=n; i++){
            scanf("%d", &k[i]);
            sk+=k[i];
        }
        ans[1]=get_ans(1);
        ans[2]=get_ans(2);
        dp[1]=ans[2]-ans[1];
        for(i=2; i<n; i++)
        dp[i]=dp[i-1]+(a+b-1)*k[i]+sk;
        for(i=3; i<=n; i++)
        ans[i]=ans[i-1]+dp[i-1];
        t=1;
        for(i=2; i<=n; i++){
            if(ans[i]<ans[t])
            t=i;
        }
        printf("%d\n", t);
    }
    return 0;
}