제출 #1082709

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1082709		Arp	Train (APIO24_train)	C++17	10 / 100	88 ms	25488 KiB

train

#include<bits/stdc++.h>
using namespace std;

using i64 = long long;
const i64 inf = 1e18;

struct Data{
	i64 cost;
	int cur,node,mask;
	Data(i64 _c=0LL,int _cu=0,int _no=0,int _ma=0){
		cost = _c;
		cur = _cu;
		node = _no;
		mask = _ma;
	}
	bool operator < (Data D) const{
		if(D.cost == cost){
			return D.cur <= cur;
		}
		return D.cost < cost;
	}
};

i64 solve(int N,int M,int W,vector<int> T,
	vector<int> X,vector<int> Y,vector<int> A,vector<int> B,vector<int> C,
	vector<int> L,vector<int> R){
	
	assert(W <= 10);

	vector<vector<int>> adj(N);
	for(int i = 0;i < M;i++){
		int u = X[i];
		adj[u].push_back(i);
	} 
	for(int i = 0;i < N;i++){
		sort(adj[i].begin(),adj[i].end(),[&](int i,int j){
			return A[i] > A[j];
		});
	}
	vector<vector<int>> cur(N,vector<int>((1 << W)));
	vector<vector<i64>> dist(M + 1,vector<i64>(1 << W,inf));
	priority_queue<Data,vector<Data>,less<Data>> pq;
	pq.push(Data(0LL,-1,0,0));

	while(!pq.empty()){
		Data d = pq.top();
		pq.pop();
		int u = (d.node == 0 ? 0 : Y[d.node - 1]);
		i64 cost = d.cost;
		int time = d.cur;
		int mask = d.mask;
		if(dist[d.node][mask] != inf) continue;
		dist[d.node][mask] = cost;
		while(cur[u][mask] < (int) adj[u].size()){
			int ind = adj[u][cur[u][mask]];
			int v = Y[ind];
			i64 w = C[ind];
			if(time <= A[ind]){
				int ma = 0; 
				for(int i = 0;i < W;i++){
					if((1 << i) & mask) continue;
					if(!(R[i] < time || L[i] > B[ind])){
						// have to consider
						if(R[i] < A[ind]){
							// have to pay
							w += T[u];
						}
						ma |= (1 << i);
					}
				}
				pq.push(Data(cost + w,B[ind],ind + 1,mask | ma));
				++ cur[u][mask];
			}else break;
		}
	}
	i64 ans = 1e18;
	for(int i = 1;i <= M;i++){
		if(Y[i - 1] != N - 1) continue;
		for(int mask = 0;mask < (1 << W);mask++){
			if(dist[i][mask] != inf){
				i64 cur = dist[i][mask];
				for(int j = 0;j < W;j++){
					if((1 << j) & mask) continue;
					cur += T[N - 1];
				}
				ans = min(ans,cur);
			}
		}
	}
	return (ans == inf ? -1 : ans);
}

컴파일 시 표준 에러 (stderr) 메시지

train.cpp: In function 'i64 solve(int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:56:8: warning: unused variable 'v' [-Wunused-variable]
   56 |    int v = Y[ind];
      |        ^

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 0 / 30
• Subtask 4: 0 / 60