Codeforces 1084A - The Fair Nut and Elevator
题解链接
题目链接
https://codeforces.com/contest/1084/problem/A
题意
有一个电梯,它初始时会停留在第 k k k 层楼,每当有一个人要从第 a a a 楼去第 b b b 楼,电梯就会始终按照 k → a → b → k k \rightarrow a \rightarrow b \rightarrow k k→a→b→k 的路线去运行,而且电梯最多只会同时容纳一个人,现在有一个 n n n 层楼的楼房,第 i i i 层有 a i a_i ai 个人,这一层的每个人每天都要按照 i → 1 → i i \rightarrow 1 \rightarrow i i→1→i 的顺序使用电梯,问将电梯的 k k k 设为多少时,每天的运行距离最少。
思路
注意到 n n n 只有 100 100 100,暴力枚举,复杂度 O ( n 2 ) O(n ^ 2) O(n2)。
实现
#include <bits/stdc++.h>
typedef long long ll;
ll a[1007], n, ans = 0x3f3f3f3f;
int main() {
scanf("%lld", &n);
for (ll i = 1; i <= n; i++) scanf("%lld", a + i);
for (ll i = 1; i <= n; i++) {
ll sum = 0;
for (ll j = 1; j <= n; j++) sum += a[j] * 2 * (abs(i - j) + abs(j - 1) + abs(1 - i));
ans = std::min(ans, sum);
}
printf("%lld\n", ans);
return 0;
}