oj.uz

답안 #458334

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
458334 2021-08-08T04:10:20 Z izhang05 Džumbus (COCI19_dzumbus) C++17
20 / 110
 60 ms 18112 KB 
#include <bits/stdc++.h>

using namespace std;

//#define DEBUG
void setIO(const string &name) {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cin.exceptions(istream::failbit);
#ifdef LOCAL
    freopen((name + ".in").c_str(), "r", stdin);
    freopen((name + ".out").c_str(), "w", stdout);
    freopen((name + ".out").c_str(), "w", stderr);
#endif
}

const int inf = 0x3f3f3f3f, mod = 1e9 + 7, maxn = 1e3 + 5;
const long long INFL = 0x3f3f3f3f3f3f3f3f;
long long d[maxn];
bool vis[maxn];
vector<int> adj[maxn];
vector<long long> dp[maxn][3];

vector<long long> ad(vector<long long> a, vector<long long> b) {
    vector<long long> res(a.size() + b.size() - 1, INFL);
    for (int i = 0; i < (int) a.size(); ++i) {
        for (int j = 0; j < (int) b.size(); ++j) {
            res[i + j] = min(res[i + j], a[i] + b[j]);
        }
    }
    return res;
}

vector<long long> mn(vector<long long> a, vector<long long> b) {
    while (a.size() < b.size()) {
        a.push_back(INFL);
    }
    for (int i = 0; i < b.size(); ++i) {
        a[i] = min(a[i], b[i]);
    }
    return a;
}


void dfs(int c, int p) {
    vis[c] = true;
    dp[c][0] = {0};
    dp[c][1] = {INFL, d[c]};
    for (auto &i : adj[c]) {
        if (i != p) {
            dfs(i, c);
            dp[c][0] = ad(dp[c][0], mn(dp[i][0], dp[i][2]));
            dp[c][2] = mn(ad(dp[c][2], mn(mn(dp[i][0], dp[i][1]), dp[i][2])), ad(dp[c][1], mn(dp[i][1], dp[i][2])));
            dp[c][1] = ad(dp[c][1], dp[i][0]);
        }
    }
}


int main() {
    setIO("3");

    int n, m;
    cin >> n >> m;
    for (int i = 0; i < n; ++i) {
        cin >> d[i];
    }
    for (int i = 0; i < m; ++i) {
        int a, b;
        cin >> a >> b;
        --a, --b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    vector<long long> res = {0};
    for (int i = 0; i < n; ++i) {
        if (!vis[i]) {
            dfs(0, -1);
            res = ad(res, mn(dp[i][0], dp[i][2]));
        }
    }

    int q;
    cin >> q;
    vector<pair<long long, int>> queries(q);
    for (int i = 0; i < q; ++i) {
        cin >> queries[i].first;
        queries[i].second = i;
    }
    sort(queries.begin(), queries.end());
    reverse(queries.begin(), queries.end());
    vector<int> sol(q);
    int j = 0;
#ifdef DEBUG
    for (int i = 0; i < n; ++i) {
        cout << i << ":\n";
        for (int k = 0; k <(int) dp[i][0].size(); ++k) {
            cout << k << " " << dp[i][0][k] << "\n";
        }
        cout << "\n";
        for (int k = 0; k < (int)dp[i][1].size(); ++k) {
            cout << k << " " << dp[i][1][k] << "\n";
        }
        cout << "\n";
    }
#endif
    for (int i = n - 1; i >= 0; --i) {
#ifdef DEBUG
        cout << i << " " << dp[0][0][i] << "\n";
#endif
        while (j < q && res[i] <= queries[j].first) {
            sol[queries[j].second] = i;
            ++j;
        }
    }
    for (auto &i : sol) {
        cout << i << "\n";
    }
    return 0;
}

Compilation message

dzumbus.cpp: In function 'std::vector<long long int> mn(std::vector<long long int>, std::vector<long long int>)':
dzumbus.cpp:38:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |     for (int i = 0; i < b.size(); ++i) {
      |                     ~~^~~~~~~~~~

Subtask #1
20.0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 9804 KB Output is correct
2 Correct 18 ms 13836 KB Output is correct

Subtask #2
0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 9804 KB Output is correct
2 Correct 18 ms 13836 KB Output is correct
3 Incorrect 60 ms 18112 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 48 ms 4964 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Correct 17 ms 9804 KB Output is correct
2 Correct 18 ms 13836 KB Output is correct
3 Incorrect 60 ms 18112 KB Output isn't correct
4 Halted 0 ms 0 KB -