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;
}

 

posted @ 2024-03-23 00:25  Gold_stein  阅读(4)  评论(0编辑  收藏  举报