oj.uz

Submission #756127

# Submission time Handle Problem Language Result Execution time Memory
756127 2023-06-11T07:34:36 Z HaciyevAlik Cyberland (APIO23_cyberland) C++17
49 / 100
 937 ms 8696 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];
struct triple {
    int u,v,w;
};
bool comp(triple a,triple b) {
    return a.u<b.u;
}
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]});
    }
    //check if it's a chain and apply all the special abilities -> m=n-1 && x[i]=i && y[i]=i+1
    if(m==n-1) {
        bool f=1;
        vector<triple> v;
        for(int i=0;i<m;++i) {
            triple cur;
            cur.u = x[i],cur.v=y[i],cur.w=c[i];
            v.pb(cur);
            if(x[i]+1!=y[i]) {
                f=0; break;
            }
        }
        if(f) {
            sort(v.begin(),v.end(),comp);
            double res=0;
            for(int i=0;i<int(v.size());++i) {
                triple cur=v[i];
                res+=cur.w;
                if(cur.v==h) {
                    break;
                }
                if(arr[cur.v]==2) {
                    res/=2;
                } else if(!arr[cur.v]) {
                    res=0;
                }
            }
            return res;
        }
    }
    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<long,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(dis[h]==1e10) {
        return -1;
    } else {
        return dis[h];
    }
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 848 ms 3276 KB Correct.
2 Correct 937 ms 3168 KB Correct.

Subtask #2
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 30 ms 3156 KB Correct.
2 Correct 36 ms 3192 KB Correct.
3 Correct 34 ms 3208 KB Correct.
4 Correct 36 ms 3176 KB Correct.
5 Correct 34 ms 3184 KB Correct.
6 Correct 28 ms 3896 KB Correct.
7 Correct 30 ms 3796 KB Correct.
8 Correct 14 ms 4620 KB Correct.
9 Correct 129 ms 3200 KB Correct.
10 Correct 115 ms 3092 KB Correct.

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 39 ms 3188 KB Correct.
2 Correct 50 ms 3180 KB Correct.
3 Correct 37 ms 3256 KB Correct.
4 Correct 126 ms 3156 KB Correct.
5 Correct 118 ms 3108 KB Correct.
6 Correct 8 ms 3924 KB Correct.

Subtask #4
0.0 / 19.0

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

Subtask #5
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 31 ms 3268 KB Correct.
2 Correct 34 ms 3156 KB Correct.
3 Correct 35 ms 3248 KB Correct.
4 Correct 25 ms 4052 KB Correct.
5 Correct 127 ms 3092 KB Correct.

Subtask #6
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 3256 KB Correct.
2 Correct 31 ms 3204 KB Correct.
3 Correct 46 ms 8696 KB Correct.
4 Correct 21 ms 4108 KB Correct.
5 Correct 129 ms 3092 KB Correct.
6 Correct 39 ms 3276 KB Correct.

Subtask #7
0.0 / 29.0

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

Subtask #8
0.0 / 3.0

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