Submission #1232794

#	Time	Username	Problem	Language	Result	Execution time	Memory
1232794		peyman_sf	Restore Array (RMI19_restore)	C++20	100 / 100	131 ms	736 KiB

restore

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<ll> vll;
typedef pair<ll, ll> pll;

const ll maxn = 5010, maxm = 1e4 + 5010 * 2;
const ll mod = 1e9 + 7;
const ll inf = 2e9;

#define TextIO ifstream fileIn("input.txt"); cin.rdbuf(fileIn.rdbuf()); ofstream fileOut("output.txt"); cout.rdbuf(fileOut.rdbuf());
#define FastIO ios_base::sync_with_stdio(false); cin.tie(NULL);
#define print(x) for (auto i : x) cout << i << ' '; cout << endl
#define out(x) {cout << x << endl; return;}
#define all(x) (x).begin(), (x).end()
#define pb push_back
#define endl '\n'

ll gcd(ll a, ll b) { if (b == 0) return a; return gcd(b, a % b);}
ll lcm(ll a, ll b) { return (a * b) / gcd(a, b); }
ll pw(ll a, ll b, ll md = mod) {ll res = 1; b = max(b, 0ll); while(b) {if (b & 1){res = (a * res) % md;} a = (a * a) % md; b >>= 1;} return (res % md);}

int n, m;
struct {int u, v, w;} E[maxm];
int d[maxn][3];
int vis[maxn];

void solve() {
    cin >> n >> m;
    for (int i = 1; i <= m; i++) {
        int l, r, k, x;
        cin >> l >> r >> k >> x;
        l++, r++;
        if (x == 1) E[i] = {r, l - 1, k - (r - l + 1) - 1};
        else E[i] = {l - 1, r, (r - l + 1) - k};
    }

    for (int i = 1; i <= n; i++) {
        E[i + m] = {i - 1, i, 1};
        E[i + m + n] = {i, i - 1, 0};
    }

    m += 2 * n;
    for (int i = 0; i <= n; i++) d[i][0] = d[i][1] = inf;
    d[0][0] = d[0][1] = 0;
    for (int j = 1; j <= m; j++)
        d[E[j].v][1] = min({d[E[j].v][0], d[E[j].u][0] + E[j].w, d[E[j].v][1]});

    for (int i = 2; i < n; i++) {
        for (ll j = 0; j <= n; j++) {
            d[j][0] = d[j][1];
            d[j][1] = inf;
        }
        d[0][1] = 0;

        for (int j = 1; j <= m; j++)
            d[E[j].v][1] = min({d[E[j].v][0], d[E[j].u][0] + E[j].w, d[E[j].v][1]});

    }

    vector<int> ans;
    for (int i = 1; i <= n; i++) {
        ans.pb(d[i][1] - d[i - 1][1]);
        if (d[i][1] - d[i - 1][1] > 1 || d[i][1] - d[i - 1][1] < 0) out(-1);
    }

    print(ans);
}

int main() {
    FastIO
    ll t = 1;
    // cin >> t;
    while (t--)
        solve();

    return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 13 / 13
• Subtask 3: 25 / 25
• Subtask 4: 55 / 55