2023 Hubei Provincial Collegiate Programming Contest TJ
最开始的思路想错了:
把所有是负数段的连成一块,发现下一步会走到这些负数区域的时候,就算出来需要差了多少,然后在当前这个正数块停留一段时间。
但有可能我们当前踩着的这一块正数非常小,导致我们要等很长时间,所以正确的贪心应该是:
记录最大值的位置,先不停往后走,当发现结果变为负数的时候,就算出需要在之前那个最大的位置停留多久,然后把时间补上。
#include <bits/stdc++.h> #define int long long using namespace std; const int N = 1e6 + 9, p = 998244353; int a[N], prefix[N]; signed main() { ios::sync_with_stdio(0), cin.tie(0), cout.tie(0); int n;cin >> n; for(int i = 1;i <= n; ++ i)cin >> a[i]; for(int i = 1;i <= n; ++ i)prefix[i] = prefix[i - 1] + a[i]; if(prefix[n] < 0) { cout << -1 << '\n'; return 0; } //遇到的第一个是负数 for(int i = 1;i <= n; ++ i) { if(prefix[i] != 0) { if(prefix[i] < 0) { cout << -1 << '\n'; return 0; } break; } } int ans = 0, res = 0, mx = 0; //前面一步步推进 for(int i = 1;i <= n; ++ i) { mx = max(mx, prefix[i]); res += prefix[i]; ans ++;//走一秒 if(res < 0)//说明走快了,应该在前面多停留一段时间的 { //补几秒钟 int ti = (-res + mx - 1) / mx; // 除以mx向上取整,所以需要+mx - 1 ans += ti; res += ti * mx; } } cout << ans << '\n'; return 0; }