Submission #524183

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
524183	2022-02-08T18:27:22 Z	Cross_Ratio	Firefighting (NOI20_firefighting)	C++14	3 / 100	273 ms	28128 KB

Firefighting

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int INF = 1e18;
typedef pair<int,int> P;
vector<vector<P>> adj;
bool type[300005];
vector<int> ans;
int K;
int dfs(int c, int p, int len) {
    int m1 = 0, m2 = 0, mi = INF;
    for(P n2 : adj[c]) {
        int n = n2.first;
        int d = n2.second;
        if(n==p) continue;
        int l = dfs(n, c, d);
        if(type[n]) mi = min(mi, l); // sobang case
        else {
            if(l >= m1) {
                m2 = m1;
                m1 = l;
            }
            else if(l > m2) m2 = l;
        }
    }
    if(mi == INF) {
        if(len == -1 || m1 + len > K || m1 + m2 > K) {
            type[c] = true;
            ans.push_back(c);
            return len;
        }
        type[c] = false;
        return len + m1;
    }
    if(mi + m1 > K) {
        type[c] = true;
        ans.push_back(c);
        return len;
    }
    if(mi + len <= K) {
        type[c] = true;
        return mi + len;
    }
    type[c] = false;
    return len;
}
signed main() {
    cin.sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int N;
    cin >> N >> K;
    int i;
    adj.resize(N);
    for(i=0;i<N-1;i++) {
        int a, b, c;
        cin >> a >> b >>c;
        adj[a-1].push_back(P(b-1, c));
        adj[b-1].push_back(P(a-1, c));
    }
    dfs(0, -1, -1);
    cout << ans.size() << '\n';
    for(i=0;i<ans.size();i++) cout << ans[i] + 1 << ' ';
}

Compilation message

Firefighting.cpp: In function 'int main()':
Firefighting.cpp:63:14: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |     for(i=0;i<ans.size();i++) cout << ans[i] + 1 << ' ';
      |             ~^~~~~~~~~~~

Subtask #1
3.0 / 3.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	270 ms	28128 KB	Output is correct
2	Correct	258 ms	28088 KB	Output is correct
3	Correct	93 ms	10652 KB	Output is correct
4	Correct	247 ms	27336 KB	Output is correct
5	Correct	0 ms	204 KB	Output is correct
6	Correct	1 ms	204 KB	Output is correct

Subtask #2
0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	204 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	0 ms	204 KB	Output is correct
4	Correct	0 ms	204 KB	Output is correct
5	Correct	0 ms	204 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	0 ms	204 KB	Output is correct
8	Correct	0 ms	204 KB	Output is correct
9	Correct	1 ms	316 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	1 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Incorrect	0 ms	204 KB	Output isn't correct
14	Halted	0 ms	0 KB	-

Subtask #3
0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	0 ms	204 KB	Output is correct
2	Correct	0 ms	204 KB	Output is correct
3	Correct	0 ms	204 KB	Output is correct
4	Correct	0 ms	204 KB	Output is correct
5	Correct	0 ms	204 KB	Output is correct
6	Correct	0 ms	204 KB	Output is correct
7	Correct	1 ms	204 KB	Output is correct
8	Correct	1 ms	208 KB	Output is correct
9	Correct	0 ms	204 KB	Output is correct
10	Correct	1 ms	204 KB	Output is correct
11	Correct	0 ms	204 KB	Output is correct
12	Correct	0 ms	204 KB	Output is correct
13	Correct	0 ms	204 KB	Output is correct
14	Correct	0 ms	204 KB	Output is correct
15	Correct	0 ms	204 KB	Output is correct
16	Correct	0 ms	204 KB	Output is correct
17	Incorrect	0 ms	204 KB	Output isn't correct
18	Halted	0 ms	0 KB	-