제출 #982460

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
982460		Kenjikrab	사이버랜드 (APIO23_cyberland)	C++17	40 / 100	3061 ms	38564 KiB

cyberland

#include "cyberland.h"
#include<bits/stdc++.h>
#define fi first 
#define se second
#define db long double
using namespace std;
int const n_max=1e5+10;
db len[n_max][31];
vector<pair<int,db>> node[n_max];
bitset<n_max> canend;
db ans=DBL_MAX;

void reset(int N,int K)
{
    for(int i=0;i<N;i++)for(int j=0;j<=K;j++)len[i][j]=LDBL_MAX;
    for(int i=0;i<N;i++)node[i].clear();
    canend.reset();
    ans=LDBL_MAX;
}
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr)
{
    reset(N,K);
    for(int i=0;i<M;i++)
    {
        node[x[i]].push_back({y[i],(db)c[i]});
        node[y[i]].push_back({x[i],(db)c[i]});
    }
    canend[0]=true;
    queue<int> qq;
    qq.push(0);
    while(!qq.empty())
    {
        int u=qq.front();
        qq.pop();
        for(auto it:node[u])
        {
            int v=it.fi;
            if(canend[v]||v==H)continue;
            canend[v]=true;
            qq.push(v);
        }
    }
    priority_queue<pair<pair<db,db>,pair<int,int>>,vector<pair<pair<db,db>,pair<int,int>>>,
    greater<pair<pair<db,db>,pair<int,int>>>> q;
    q.push({{0,1},{K,H}});
    for(int i=0;i<=K;i++)len[H][i]=0;
    while(!q.empty())
    {
        db l=q.top().fi.fi,cost=q.top().fi.se;
        int k=q.top().se.fi,u=q.top().se.se;
        q.pop();
        //cout<<l<<" "<<cost<<" "<<k<<" "<<u<<endl;
        if(l>ans)continue;
        if(len[u][k]<l)continue;
        if(u==H&&l!=0)continue;
        for(auto it:node[u])
        {
            int v=it.fi;
            db nl=it.se*cost+l;
            if(v==0)ans=min(ans,nl);
            if(arr[v]==0)
            {
                if(len[v][k]>nl)
                {
                    for(int i=k;i<=K;i++)len[v][i]=nl;
                    if(canend[v])ans=min(ans,nl);
                    continue;
                }
            }
            else if(arr[v]==2&&k>=1)
            {
                if(len[v][k-1]>nl)
                {
                    for(int i=k-1;i<=K;i++)len[v][i]=nl;
                    q.push({{nl,cost/2},{k-1,v}});
                    //cout<<v<<endl;
                }
                continue;
            }
            else
            {
                if(len[v][k]>nl)
                {
                    for(int i=k;i<=K;i++)len[v][i]=nl;
                    q.push({{nl,cost},{k,v}});
                    //cout<<v<<endl;
                }
            }
        }

    }
    return ans;
}

• Subtask 1: 0 / 5
• Subtask 2: 8 / 8
• Subtask 3: 13 / 13
• Subtask 4: 19 / 19
• Subtask 5: 0 / 7
• Subtask 6: 0 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3