Submission #497932

#	Time	Username	Problem	Language	Result	Execution time	Memory
497932		vinnipuh01	Bigger segments (IZhO19_segments)	C++17	100 / 100	73 ms	16060 KiB

segments

/**
 * LES GREATEABLES BRO TEAM
**/
 
#include <bits/stdc++.h>
 
using namespace std;
 
using ll = long long;
#define sz(x) (int)x.size()
const bool FLAG = false;
void setIO(const string &f = "");
#define int ll
const int N = 5e5 + 137;
ll a[N], dp[N], last[N], pref[N];
 
void solve() {
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        pref[i] = a[i];
        pref[i] += pref[i - 1];
    }
 
    dp[1] = 1;
    last[1] = 1;
    for (int j = 1; j <= n; j++) {
        int right = 2 * pref[j] - pref[last[j] - 1];
 
        int lo = j, hi = n, res = -1;
        while (lo < hi) {
            int mid = (lo + hi) / 2;
 
            if (pref[mid] >= right) {
                hi = mid;
            } else
                lo = mid + 1;
        }
        res = lo;
        if (pref[ res ] >= right) {
            if (dp[res] <= dp[j] + 1 ) {
                dp[res] = dp[j] + 1;
                last[res] = j + 1;
            }
        }
        if (dp[j + 1] < dp[j] || ( dp[ j + 1 ] == dp[ j ] && last[ j + 1 ] < last[ j ] )) {
            dp[j + 1] = dp[j];
            last[j + 1] = last[j];
        }
    }
    cout << dp[n];
}
 
signed main() {
    setIO();
    
    int tt = 1;
    if (FLAG) {
        cin >> tt;
    }
    while (tt--) {
        solve();
    }
    
    return 0;
}
 
void setIO(const string &f) {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    if (fopen((f + ".in").c_str(), "r")) {
        freopen((f + ".in").c_str(), "r", stdin);
        freopen((f + ".out").c_str(), "w", stdout);
    }
}

Compilation message (stderr)

segments.cpp: In function 'void setIO(const string&)':
segments.cpp:73:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   73 |         freopen((f + ".in").c_str(), "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
segments.cpp:74:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   74 |         freopen((f + ".out").c_str(), "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 13 / 13
• Subtask 2: 14 / 14
• Subtask 3: 10 / 10
• Subtask 4: 36 / 36
• Subtask 5: 27 / 27