Submission #982097

#TimeUsernameProblemLanguageResultExecution timeMemory
982097vjudge1사이버랜드 (APIO23_cyberland)C++17
0 / 100
46 ms41696 KiB
#include <bits/stdc++.h>
#define rep(a,b,c) for(int a=b; a<c; a++)
#define repa(a,b) for(auto a:b)
#define pll pair<ll, ll>
#define fi first
#define se second
#define state array<double, 3>
#define pb push_back
#define ll long long

using namespace std;

const double inf=1e18;

double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr){
	vector<pll> adj[N];
	rep(i,0,M){
		adj[x[i]].pb({y[i],c[i]});
		adj[y[i]].pb({x[i],c[i]});
	}
    K=min(K,200);
	double dis[N][K]{};
	rep(i,0,N) rep(j,0,K+1) dis[i][j]=inf;
	priority_queue<state, vector<state>, greater<state>> pq;
	pq.push({0,0,0});
	dis[0][0]=0;
	while(pq.size()){
		int u=pq.top()[2], k=pq.top()[0];
		double w=pq.top()[1];
		pq.pop();
		if(dis[u][k]<w || u==H) continue;
		//cout<<u<<" "<<k<<" "<<w<<endl;
		if(arr[u]==2){
			repa(v,adj[u]){
				if(v.fi!=H && k<K && dis[u][k+1]>(dis[u][k]+v.se+v.se)/2){
					dis[u][k+1]=(dis[u][k]+v.se+v.se)/2;
					pq.push({k+1,dis[u][k+1],u});
				}
			}
		}
		repa(v,adj[u]){
			if(dis[v.fi][k]>dis[u][k]+v.se){
				dis[v.fi][k]=dis[u][k]+v.se;
				pq.push({k,dis[v.fi][k],v.fi});
			}
			if(k<K && arr[v.fi]==2 && dis[v.fi][k+1]>(dis[u][k]+v.se)/2){
				dis[v.fi][k+1]=(dis[u][k]+v.se)/2;
				pq.push({k+1,dis[v.fi][k+1],v.fi});
			}
			if(!arr[v.fi] && dis[v.fi][k]){
				dis[v.fi][k]=0;
				pq.push({k,dis[v.fi][k],v.fi});
			}
		}
	}
	double ans=inf;
	rep(i,0,K+1) ans=min(ans,dis[H][i]);
	/*rep(i,0,N){
		cout<<i<<": ";
		rep(j,0,K) cout<<dis[i][j]<<" ";
		cout<<endl;
	}*/
	if(ans==inf) return -1;
	else return ans;
}

Compilation message (stderr)

cyberland.cpp: In function 'double solve(int, int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
cyberland.cpp:37:16: warning: narrowing conversion of '(k + 1)' from 'int' to 'double' [-Wnarrowing]
   37 |      pq.push({k+1,dis[u][k+1],u});
      |               ~^~
cyberland.cpp:37:31: warning: narrowing conversion of 'u' from 'int' to 'double' [-Wnarrowing]
   37 |      pq.push({k+1,dis[u][k+1],u});
      |                               ^
cyberland.cpp:44:14: warning: narrowing conversion of 'k' from 'int' to 'double' [-Wnarrowing]
   44 |     pq.push({k,dis[v.fi][k],v.fi});
      |              ^
cyberland.cpp:5:12: warning: narrowing conversion of 'v.std::pair<long long int, long long int>::first' from 'long long int' to 'double' [-Wnarrowing]
    5 | #define fi first
      |            ^
cyberland.cpp:44:31: note: in expansion of macro 'fi'
   44 |     pq.push({k,dis[v.fi][k],v.fi});
      |                               ^~
cyberland.cpp:48:15: warning: narrowing conversion of '(k + 1)' from 'int' to 'double' [-Wnarrowing]
   48 |     pq.push({k+1,dis[v.fi][k+1],v.fi});
      |              ~^~
cyberland.cpp:5:12: warning: narrowing conversion of 'v.std::pair<long long int, long long int>::first' from 'long long int' to 'double' [-Wnarrowing]
    5 | #define fi first
      |            ^
cyberland.cpp:48:35: note: in expansion of macro 'fi'
   48 |     pq.push({k+1,dis[v.fi][k+1],v.fi});
      |                                   ^~
cyberland.cpp:52:14: warning: narrowing conversion of 'k' from 'int' to 'double' [-Wnarrowing]
   52 |     pq.push({k,dis[v.fi][k],v.fi});
      |              ^
cyberland.cpp:5:12: warning: narrowing conversion of 'v.std::pair<long long int, long long int>::first' from 'long long int' to 'double' [-Wnarrowing]
    5 | #define fi first
      |            ^
cyberland.cpp:52:31: note: in expansion of macro 'fi'
   52 |     pq.push({k,dis[v.fi][k],v.fi});
      |                               ^~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...