Solution
条件的意思是只要你往回走,那么就要把之前没有救的村庄给救了。
考虑认识怎么走的,必定是走一段,往回走,再走一段,再往回,知道救完。
那么我们设
f
[
i
]
f[i]
f[i]表示人现在在
i
i
i,并且
1
−
i
1-i
1−i的村庄都救完的最小代价。那么我们枚举一个
j
j
j,表示人现在
j
j
j,有
f
[
j
]
f[j]
f[j]的代价,然后再从
j
+
1
j+1
j+1往返走到
i
i
i。就是说,枚举转头的地方。那么设这个代价为
w
[
i
]
[
j
]
w[i][j]
w[i][j],表示从i走到j,然后再从
j
j
j走到
i
i
i的代价,这时候我们假定
1
−
(
i
−
1
)
1-(i-1)
1−(i−1)都已经被救了了,
i
−
n
i-n
i−n没有被救。
一开始想错了,想枚举
j
j
j什么时候救,然后发现不可做,应该枚举
i
i
i什么时候救。
但是还有一个问题,就是我有可能从 i i i走到 j j j不掉头,表示了吗?答案是肯定的,考虑用 i − 1 i-1 i−1去更新 i i i的时候,就相当于不掉头。
Code
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 3e3 + 10;
int n , a[N];
ll w[N][N] , f[N] , sum[N];
int main(){
cin >> n;
for (int i = 1;i <= n;i++){
scanf("%lld",&a[i]);
sum[i] = sum[i-1] + a[i];
}
for (int i = 1;i <= n;i++){
w[i][i] = sum[n] - sum[i];
}
for (int len = 2;len <= n;len++){
int days = 3 * len - 5;
for (int i = 1;i <= n - len + 1;i ++){
int j = i + len - 1;
w[i][j] = w[i+1][j] + 2ll * (sum[n] - sum[i]) + sum[n] - sum[j];
w[i][j] = min(w[i][j] , w[i+1][j] + sum[n] - sum[i-1] + 1ll * days * a[i] + a[i] + 2ll * ( sum[n] - sum[j] ) );
}
}
memset(f , 0x3f , sizeof f);
f[0] = 0;f[1] = sum[n] - sum[1];
for (int i = 2;i <= n;i++){
f[i] = w[1][i] + 1ll * ( i - 1 ) * (sum[n] - sum[i]);
for (int j = 1;j + 1 <= i;j++){
f[i] = min(f[i] , f[j] + sum[n] - sum[j] + w[j+1][i] + 1ll * ( i - j - 1) * ( sum[n] - sum[i] ) );
}
}
cout << f[n];
return 0;
}