Submission #563231

# Submission time Handle Problem Language Result Execution time Memory
563231 2022-05-16T14:30:33 Z ngpin04 Road Closures (APIO21_roads) C++14
5 / 100
69 ms 6828 KB
#include <bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define TASK ""
#define bit(x) (1LL << (x))
#define getbit(x, i) (((x) >> (i)) & 1)
#define ALL(x) (x).begin(), (x).end() 
using namespace std;
template <typename T1, typename T2> bool mini(T1 &a, T2 b) {
    if (a > b) {a = b; return true;} return false;
}
template <typename T1, typename T2> bool maxi(T1 &a, T2 b) {
    if (a < b) {a = b; return true;} return false;
}
mt19937_64 rd(chrono::steady_clock::now().time_since_epoch().count());

int rand(int l, int r) {
    return l + rd() % (r - l + 1);
}
const int N = 1e5 + 5; 
const int oo = 1e9;
const long long ooo = 1e18;
const int mod = 1e9 + 7; // 998244353;
const long double pi = acos(-1);

#include "roads.h"

long long dp[N][2];

int fr[N];
int to[N];
int w[N];
int n;

vector <long long> sub1() {
    sort(w, w + (n - 1), greater <int>());
    long long tot = accumulate(w, w + (n - 1), 0LL);
    vector <long long> res;
    res.push_back(tot);
    for (int i = 0; i < n - 1; i++)
        tot -= w[i], res.push_back(tot);
    return res;
}

vector <long long> sub2() {
    vector <long long> res;
    res.push_back(accumulate(w, w + (n - 1), 0LL));
    for (int i = 0; i < n; i++)
    for (int j = 0; j < 2; j++)
        dp[i][j] = ooo;

    dp[0][0] = 0;
    for (int i = 1; i < n; i++) {
        dp[i][0] = min(dp[i - 1][0], dp[i - 1][1]) + w[i - 1];
        dp[i][1] = dp[i - 1][0];
    }
    res.push_back(min(dp[n - 1][0], dp[n - 1][1]));
    return res;
}

vector<long long> minimum_closure_costs(int N, vector<int> U,
                                             vector<int> V,
                                             vector<int> W) {

    bool cond2 = true;
    n = N;
    for (int i = 0; i < n - 1; i++) {
        fr[i] = U[i];
        to[i] = V[i];
        w[i] = W[i];
        cond2 &= fr[i] == i && to[i] == i + 1;
    }
    if (*max_element(ALL(U)) == 0)  
        return sub1();

    if (cond2)
        return sub2();
    return vector<long long>(N, 0);
}

//#include "grader.cpp"
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 2 ms 468 KB Output is correct
3 Correct 2 ms 468 KB Output is correct
4 Correct 2 ms 468 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 308 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 27 ms 3680 KB Output is correct
13 Correct 41 ms 5940 KB Output is correct
14 Correct 48 ms 6068 KB Output is correct
15 Correct 69 ms 6644 KB Output is correct
16 Correct 50 ms 6708 KB Output is correct
17 Correct 46 ms 6828 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 36 ms 5448 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Incorrect 30 ms 6268 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Incorrect 1 ms 212 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Incorrect 1 ms 212 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 35 ms 5700 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 35 ms 5700 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 2 ms 468 KB Output is correct
3 Correct 2 ms 468 KB Output is correct
4 Correct 2 ms 468 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 308 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 27 ms 3680 KB Output is correct
13 Correct 41 ms 5940 KB Output is correct
14 Correct 48 ms 6068 KB Output is correct
15 Correct 69 ms 6644 KB Output is correct
16 Correct 50 ms 6708 KB Output is correct
17 Correct 46 ms 6828 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 36 ms 5448 KB Output is correct
20 Correct 1 ms 212 KB Output is correct
21 Incorrect 30 ms 6268 KB Output isn't correct
22 Halted 0 ms 0 KB -