oj.uz

Submission #756012

# Submission time Handle Problem Language Result Execution time Memory
756012 2023-06-10T21:07:23 Z HaciyevAlik Cyberland (APIO23_cyberland) C++17
42 / 100
 978 ms 8736 KB 
#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define ff first
#define ss second
const int mx=1e5+5;
int n,m,h,k,used[mx],used2[15];
long dis[mx];
vector<pair<int,int>> g[mx];
void dfs(int u) {
    used[u]=1;
    for(int i=0;i<int(g[u].size());++i) {
        int v=g[u][i].ff;
        if(!used[v]&&v!=h) dfs(v);
    }
}
void dfs2(int u) {
    used2[u]=1;
    for(int i=0;i<int(g[u].size());++i) {
        int v=g[u][i].ff;
        if(!used2[v]) dfs2(v);
    }
}
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
    n=N,m=M,h=H,k=K;
    for(int i=0;i<mx;++i) g[i].clear();
    memset(used,0,sizeof(used));
    for(int i=0;i<m;++i) {
        g[x[i]].pb({y[i],c[i]});
        g[y[i]].pb({x[i],c[i]});
    }
    if(n==2||n==3) {
        memset(used2,0,sizeof(used2));
        dfs2(0);
        if(!used2[h]) {
            return -1;
        }
        if(g[0].size()==1) {
            if(g[0][0].ff==h) {
                return g[0][0].ss;
            } else {
                double res=g[0][0].ss;
                int cur=g[0][0].ff;
                if(!arr[cur]) {
                    res=0;
                } else if(arr[cur]==2 && k>0) {
                    res/=2;
                }
                res+=g[h][0].ss;
                return res;
            }
        } else {
            if(g[0][0].ff==h) {
                double res1=g[0][0].ss,res2=g[0][1].ss;
                int cur=g[0][1].ff;
                if(g[cur].size()==1) return res1;
                if(!arr[cur]) {
                    res2=0;
                } else if(arr[cur]==2&&k>0) {
                    res2/=2;
                }
                if(g[h][0].ff==0) {
                    res2+=g[h][1].ss;
                } else {
                    res2+=g[h][0].ss;
                }
                return min(res1,res2);
            } else {
                double res1=g[0][1].ss,res2=g[0][0].ss;
                int cur=g[0][0].ff;
                if(g[cur].size()==1) return res1;
                if(!arr[cur]) {
                    res2=0;
                } else if(arr[cur]==2&&k>0) {
                    res2/=2;
                }
                if(g[h][0].ff==0) {
                    res2+=g[h][1].ss;
                } else {
                    res2+=g[h][0].ss;
                }
                return min(res1,res2);
            }
        }
    }
    set<int> s;
    dfs(0);
    for(int i=0;i<n;++i) {
        if(used[i]&&!arr[i]) s.insert(i);
    }
    memset(used,0,sizeof(used));
    priority_queue<pair<int,int>> pq;
    for(int i=0;i<n;++i) dis[i]=1e10;
    for(auto i:s) {
        pq.push({0,i});
        dis[i]=0;
    }
    pq.push({0,0});
    dis[0]=0;
    while(!pq.empty()) {
        int u=pq.top().ss; pq.pop();
        if(used[u]) continue;
        used[u]=1;
        for(int i=0;i<int(g[u].size());++i) {
            int v=g[u][i].ff,w=g[u][i].ss;
            if(dis[u]+w<dis[v]) {
                dis[v]=dis[u]+w;
                pq.push({-dis[v],v});
            }
        }
    }
    if(!used[h]) {
        return -1;
    } else {
        return dis[h];
    }
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 791 ms 3260 KB Correct.
2 Correct 978 ms 3144 KB Correct.

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 30 ms 3140 KB Correct.
2 Correct 37 ms 3176 KB Correct.
3 Correct 33 ms 3152 KB Correct.
4 Correct 35 ms 3168 KB Correct.
5 Correct 33 ms 3176 KB Correct.
6 Correct 24 ms 3824 KB Correct.
7 Correct 28 ms 3796 KB Correct.
8 Correct 13 ms 4500 KB Correct.
9 Correct 171 ms 3076 KB Correct.
10 Correct 147 ms 3092 KB Correct.

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 41 ms 3156 KB Correct.
2 Correct 39 ms 3192 KB Correct.
3 Correct 37 ms 3156 KB Correct.
4 Correct 125 ms 3096 KB Correct.
5 Correct 138 ms 3108 KB Correct.
6 Correct 7 ms 3796 KB Correct.

Subtask #4
0.0 / 19.0

# Verdict Execution time Memory Grader output
1 Incorrect 39 ms 8052 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #5
0.0 / 7.0

# Verdict Execution time Memory Grader output
1 Incorrect 29 ms 3248 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #6
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 35 ms 3256 KB Correct.
2 Correct 32 ms 3276 KB Correct.
3 Correct 40 ms 8736 KB Correct.
4 Correct 19 ms 4052 KB Correct.
5 Correct 132 ms 3080 KB Correct.
6 Correct 35 ms 3232 KB Correct.

Subtask #7
0.0 / 29.0

# Verdict Execution time Memory Grader output
1 Incorrect 35 ms 3148 KB Wrong Answer.
2 Halted 0 ms 0 KB -

Subtask #8
0.0 / 3.0

# Verdict Execution time Memory Grader output
1 Incorrect 36 ms 3248 KB Wrong Answer.
2 Halted 0 ms 0 KB -