我自己的思路,可能不是最优
1.先以一个点为终点(假设以1点),这里要求出从左边到1的各点物资数量和,和右边到1点的物资数量和,以及总花费
2.枚举每个点,假设从1移动到2,那么原先1右边的所有距离就要+1,左边-1,所以我们将总花费减去左边物资和,加上右边物资和,在加上距离为1的1点物资,此时更新完后(如果是奇数点,最远距离点会多加一次,偶数多加两次),需要减去多加的物资数。
3.然后更新一下左右区间的物资数量和。
只用枚举n个点即可,时间复杂度O(n)
主要数据要开long long
#include <iostream>
using namespace std;
const int N = 500010;
typedef long long LL;
LL q[N];
int main()
{
int n;
cin >> n;
for(int i = 1; i <= n ; i ++)
{
cin >> q[i];
}
LL resl = 0,resr = 0,ans = 0;
for(int i = n % 2 != 0 ? n / 2 + 1 : n / 2,j = i;i > 1; i --)
{
LL x = q[i],y = q[ j % n + 1 ];
resl += x,resr += y;//求出左右区间物资和
ans += resl + resr;//总花费
j++;
}
if(n % 2 == 0) resr += q[n],ans +=resr;//偶数会多出一个点,随便归到一个区间
int m = 2;
LL mmin = ans;
for(int i = 2; i <= n; i ++)
{
LL h;
int j;
if(n % 2 == 0) j = m - 1;
else j = m;
if((j + n / 2) % n == 0) h = q[n];//最远距离的物资
else h = q[(j + n / 2 ) % n];
if(n % 2 == 0)
ans += resr + q[m - 1] - resl - 2 * h; // 更新总花费
else ans += resr + q[m - 1] - resl - h;
resr += q[m - 1] - h,resl = resl - q[m] + h;//更新左右区间
mmin = min(ans,mmin);
m ++;
}
cout << mmin << endl;
return 0;
}