QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#639197#9230. Routing K-Codespropane#WA 104ms43108kbC++206.4kb2024-10-13 18:15:222024-10-13 18:15:22

Judging History

你现在查看的是最新测评结果

  • [2024-10-13 18:15:22]
  • 评测
  • 测评结果:WA
  • 用时:104ms
  • 内存:43108kb
  • [2024-10-13 18:15:22]
  • 提交

answer

#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<array>
#include<algorithm>
#include<map>
using namespace std;
using LL = long long;

int main(){

#ifdef LOCAL
    freopen("data.in", "r", stdin);
    freopen("data.out", "w", stdout);
#endif

    cin.tie(0);
    cout.tie(0);
    ios::sync_with_stdio(0);

    int n, m;
    cin >> n >> m;
    if (m != n - 1){
        cout << "NIE" << '\n';
        return 0;
    }
    vector<vector<int> > g(n + 1);
    for(int i = 0; i < n - 1; i++){
        int a, b;
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }

    for(int i = 1; i <= n; i++){
        if (g[i].size() > 3){
            cout << "NIE" << '\n';
            return 0;
        }
    }

    auto bfs = [&](int st){
        queue<int> q;
        vector<int> d(n + 1, -1);
        d[st] = 0;
        q.push(st);
        while(!q.empty()){
            int t = q.front();
            q.pop();
            for(auto j : g[t]){
                if (d[j] == -1){
                    d[j] = d[t] + 1;
                    q.push(j);
                }
            }
        }
        return d;
    };

    auto d1 = bfs(1);
    int p = max_element(d1.begin(), d1.end()) - d1.begin();
    auto d2 = bfs(p);
    int q = max_element(d2.begin(), d2.end()) - d2.begin();
    auto d3 = bfs(q);

    int root = -1;
    for(int i = 1; i <= n; i++){
        if (max(d2[i], d3[i]) <= 31){
            root = i;
            break;
        }
    }

    if (root == -1){
        cout << "NIE" << '\n';
        return 0;
    }

    const LL INF = 1e18;
    LL ans = INF;

    vector<array<int, 32> > dp(n + 1);
    vector<LL> f(n + 1, INF);

    auto dfs1 = [&](auto &&dfs1, int u, int fa) -> void {
        dp[u][0] = 1;
        vector<int> sons;
        for(auto j : g[u]){
            if (j == fa) continue;
            dfs1(dfs1, j, u);
            sons.push_back(j);
            for(int i = 0; i + 1 <= 31; i++){
                dp[u][i + 1] += dp[j][i];
            }
        }
        if (sons.size() <= 2){
            if (sons.size() == 0){
                f[u] = 1;
            }
            else if (sons.size() == 1){
                f[u] = 1 + f[sons[0]];
                for(int i = 0; i + 1 <= 31; i++){
                    f[u] += (1LL << i) * dp[sons[0]][i];
                }
            }
            else{
                LL mn = 1e18;
                for(auto _ : {0, 1}){
                    swap(sons[0], sons[1]);
                    LL sum = 1 + f[sons[0]] + f[sons[1]];
                    for(int i = 0; i < 2; i++){
                        int j = sons[i];
                        for(int k = 0; k + 1 <= 31; k++){
                            sum += (1 + i) * (1LL << k) * dp[j][k];
                        }
                    }
                    mn = min(mn, sum);
                }
                f[u] = mn;
            }
        }
        f[u] = min(f[u], INF);
    };

    dfs1(dfs1, root, -1);

    auto dfs2 = [&](auto &&dfs2, int u, int fa, array<int, 32> pre, LL val) -> void {
        map<int, array<int, 32> > ndp;
        map<int, int> nf;
        {
            int leaf = -1;
            vector<int> sons;
            for(auto j : g[u]){
                if (j == fa){
                    ndp[j] = pre;
                    nf[j] = val;
                }
                else{
                    ndp[j] = dp[j];
                    nf[j] = f[j];
                }
                if (g[j].size() == 1 and leaf == -1){
                    leaf = j;
                }
                else{
                    sons.push_back(j);
                }
            }
            
            if (sons.size() <= 2){
                LL mn = 1e18;
                if (sons.size() == 0){
                    mn = 1;
                }
                else if (sons.size() == 1){
                    mn = 1 + nf[sons[0]];
                    for(int i = 0; i + 1 <= 31; i++){
                        mn += (1LL << i) * ndp[sons[0]][i];
                    }
                }
                else{
                    for(auto _ : {0, 1}){
                        swap(sons[0], sons[1]);
                        LL sum = 1 + nf[sons[0]] + nf[sons[1]];
                        for(int i = 0; i < 2; i++){
                            int j = sons[i];
                            for(int k = 0; k + 1 <= 31; k++){
                                sum += (1 + i) * (1LL << k) * ndp[j][k];
                            }
                        }
                        mn = min(mn, sum);
                    }
                }
                // cout << u << ' ' << mn << '\n';
                ans = min(ans, mn);
            }
        }        
        for(auto j : g[u]){
            if (j == fa) continue;
            if (max(d2[j], d3[j]) > 31) continue;
            array<int, 32> cnt{};
            cnt[0] = 1;
            vector<int> sons;
            for(auto k : g[u]){
                if (k != j){
                    sons.push_back(k);
                    for(int i = 0; i + 1 <= 31; i++){
                        cnt[i + 1] += ndp[k][i];
                    }
                }
            }
            LL mn = 1e18;
            if (sons.size() <= 2){
                if (sons.size() == 0){
                    mn = 1;
                }
                else if (sons.size() == 1){
                    mn = 1 + nf[sons[0]];
                    for(int i = 0; i + 1 <= 31; i++){
                        mn += (1LL << i) * ndp[sons[0]][i];
                    }
                }
                else{
                    for(auto _ : {0, 1}){
                        swap(sons[0], sons[1]);
                        LL sum = 1 + nf[sons[0]] + nf[sons[1]];
                        for(int i = 0; i < 2; i++){
                            int j = sons[i];
                            for(int k = 0; k + 1 <= 31; k++){
                                sum += (1 + i) * (1LL << k) * ndp[j][k];
                            }
                        }
                        mn = min(mn, sum);
                    }
                }
            }
            mn = min(mn, INF);
            dfs2(dfs2, j, u, cnt, mn);
        }

    };

    dfs2(dfs2, root, -1, {}, 0);

    if (ans == INF) cout << "NIE" << '\n';
    else cout << ans << '\n';


}

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 3504kb

input:

4 3
1 2
1 3
1 4

output:

6

result:

ok single line: '6'

Test #2:

score: 0
Accepted
time: 0ms
memory: 3868kb

input:

4 6
1 2
2 3
3 4
4 1
1 3
2 4

output:

NIE

result:

ok single line: 'NIE'

Test #3:

score: 0
Accepted
time: 0ms
memory: 3624kb

input:

10 10
9 3
5 10
1 4
10 8
8 3
7 3
9 6
8 6
7 2
4 5

output:

NIE

result:

ok single line: 'NIE'

Test #4:

score: 0
Accepted
time: 54ms
memory: 16484kb

input:

200000 199999
172774 188052
195063 155577
38023 148303
30605 155047
170238 19344
46835 58255
154268 109062
197059 116041
136424 12058
42062 149807
109545 115380
132322 51106
16706 162612
62234 31319
195735 80435
173898 84051
19876 102910
186924 9136
123094 117097
145054 173049
137364 152731
82662 18...

output:

NIE

result:

ok single line: 'NIE'

Test #5:

score: -100
Wrong Answer
time: 104ms
memory: 43108kb

input:

200000 199999
168192 30733
164413 46673
175210 2329
69389 67102
33991 152553
91061 55265
166724 117726
90148 157176
34831 12213
60756 105488
121891 155192
82202 155666
102083 156661
38968 75200
190004 107321
72436 92732
64314 65004
39210 91106
70455 173568
179742 29844
126232 19552
133509 110602
131...

output:

10478835005629

result:

wrong answer 1st lines differ - expected: '20980030044349', found: '10478835005629'