Submission #981978

#	Time	Username	Problem	Language	Result	Execution time	Memory
981978		vjudge1	Cyberland (APIO23_cyberland)	C++17	0 / 100	3034 ms	53044 KiB

cyberland

#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){
	if(K>30) return -1;
	vector<pll> adj[N];
	rep(i,0,M){
		adj[x[i]].pb({y[i],c[i]});
		adj[y[i]].pb({x[i],c[i]});
	}
	double dis[N][35]{};
	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()[1], k=pq.top()[2];
		double w=pq.top()[0];
		pq.pop();
		if(dis[u][k]<w || u==H) continue;
		cout<<u<<" "<<k<<" "<<w<<endl;
		if(arr[u]==2){
			repa(v,adj[u]){
				if(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({dis[u][k+1],u,k+1});
				}
			}
		}
		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({dis[v.fi][k],v.fi,k});
			}
			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({dis[v.fi][k+1],v.fi,k+1});
			}
			if(!arr[v.fi] && dis[v.fi][k]){
				dis[v.fi][k]=0;
				pq.push({dis[v.fi][k],v.fi,k});
			}
		}
	}
	double ans=inf;
	rep(i,0,K) 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:27: warning: narrowing conversion of 'u' from 'int' to 'double' [-Wnarrowing]
   37 |      pq.push({dis[u][k+1],u,k+1});
      |                           ^
cyberland.cpp:37:30: warning: narrowing conversion of '(k + 1)' from 'int' to 'double' [-Wnarrowing]
   37 |      pq.push({dis[u][k+1],u,k+1});
      |                             ~^~
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:29: note: in expansion of macro 'fi'
   44 |     pq.push({dis[v.fi][k],v.fi,k});
      |                             ^~
cyberland.cpp:44:32: warning: narrowing conversion of 'k' from 'int' to 'double' [-Wnarrowing]
   44 |     pq.push({dis[v.fi][k],v.fi,k});
      |                                ^
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:31: note: in expansion of macro 'fi'
   48 |     pq.push({dis[v.fi][k+1],v.fi,k+1});
      |                               ^~
cyberland.cpp:48:35: warning: narrowing conversion of '(k + 1)' from 'int' to 'double' [-Wnarrowing]
   48 |     pq.push({dis[v.fi][k+1],v.fi,k+1});
      |                                  ~^~
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:29: note: in expansion of macro 'fi'
   52 |     pq.push({dis[v.fi][k],v.fi,k});
      |                             ^~
cyberland.cpp:52:32: warning: narrowing conversion of 'k' from 'int' to 'double' [-Wnarrowing]
   52 |     pq.push({dis[v.fi][k],v.fi,k});
      |                                ^

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