#982353 (C++17) No.2764 Warp Drive Spacecraft

提出ソース

結果

問題	No.2764 Warp Drive Spacecraft
ユーザー	沙耶花
提出日時	2024-05-17 22:29:22
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	508 ms / 3,000 ms
コード長	2,547 bytes
コンパイル時間	4,655 ms
コンパイル使用メモリ	276,856 KB
実行使用メモリ	26,156 KB
最終ジャッジ日時	2024-07-17 20:23:48
合計ジャッジ時間	14,423 ms
ジャッジサーバーID （参考情報）	judge3 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	1 ms 6,944 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	1 ms 6,944 KB
testcase_07	AC	1 ms 6,940 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	1 ms 6,944 KB
testcase_10	AC	3 ms 6,944 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	3 ms 6,944 KB
testcase_13	AC	3 ms 6,944 KB
testcase_14	AC	2 ms 6,944 KB
testcase_15	AC	2 ms 6,944 KB
testcase_16	AC	262 ms 21,912 KB
testcase_17	AC	256 ms 22,104 KB
testcase_18	AC	253 ms 22,160 KB
testcase_19	AC	378 ms 20,168 KB
testcase_20	AC	390 ms 21,444 KB
testcase_21	AC	392 ms 20,384 KB
testcase_22	AC	391 ms 21,372 KB
testcase_23	AC	386 ms 20,064 KB
testcase_24	AC	386 ms 21,324 KB
testcase_25	AC	394 ms 21,528 KB
testcase_26	AC	498 ms 25,332 KB
testcase_27	AC	482 ms 25,080 KB
testcase_28	AC	476 ms 25,620 KB
testcase_29	AC	489 ms 25,684 KB
testcase_30	AC	481 ms 25,064 KB
testcase_31	AC	508 ms 26,044 KB
testcase_32	AC	499 ms 26,156 KB
testcase_33	AC	235 ms 15,444 KB
testcase_34	AC	236 ms 15,324 KB
testcase_35	AC	238 ms 15,476 KB
testcase_36	AC	241 ms 22,152 KB
testcase_37	AC	243 ms 19,488 KB
testcase_38	AC	208 ms 17,536 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000001
/**
 * Author: Simon Lindholm
 * Date: 2017-04-20
 * License: CC0
 * Source: own work
 * Description: Container where you can add lines of the form kx+m, and query maximum values at points x.
 *  Useful for dynamic programming (``convex hull trick'').
 * Time: O(\log N)
 * Status: stress-tested
 */
using ll =long long;
struct Line {
	mutable ll k, m, p;
	bool operator<(const Line& o) const { return k < o.k; }
	bool operator<(ll x) const { return p < x; }
};

struct LineContainer : multiset<Line, less<>> {
	// (for doubles, use inf = 1/.0, div(a,b) = a/b)
	static const ll inf = LLONG_MAX - 1000000000LL;
	ll div(ll a, ll b) { // floored division
		return a / b - ((a ^ b) < 0 && a % b); }
	bool isect(iterator x, iterator y) {
		if (y == end()) return x->p = inf, 0;
		if (x->k == y->k) x->p = x->m > y->m ? inf : -inf;
		else x->p = div(y->m - x->m, x->k - y->k);
		return x->p >= y->p;
	}
	void add(ll k, ll m) {
		auto z = insert({k, m, 0}), y = z++, x = y;
		while (isect(y, z)) z = erase(z);
		if (x != begin() && isect(--x, y)) isect(x, y = erase(y));
		while ((y = x) != begin() && (--x)->p >= y->p)
			isect(x, erase(y));
	}
	ll query(ll x) {
		assert(!empty());
		auto l = *lower_bound(x);
		return l.k * x + l.m;
	}
};

vector<long long> get_distance(vector<vector<pair<int,long long>>> &E,int s){
	int N = E.size();
	vector<long long> dist(N,Inf64);
	dist[s] = 0;
	priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<pair<long long,int>>> pq;
	pq.push({0,s});
	while(!pq.empty()){
		auto [d,u] = pq.top();
		pq.pop();
		if(dist[u] !=d )continue;
		for(auto [v,w]:E[u]){
			if(dist[v] > dist[u] + w){
				dist[v] = dist[u] + w;
				pq.push({dist[v],v});
			}
		}
	}
	return dist;
}

int main(){
	int N,M;
	cin>>N>>M;
	vector<long long> w(N);
	rep(i,N)cin>>w[i];
	
	vector<vector<pair<int,long long>>> E(N);
	rep(i,M){
		int u,v;
		long long ww;
		cin>>u>>v>>ww;
		u--,v--;
		E[u].emplace_back(v,ww);
		E[v].emplace_back(u,ww);
		//E[v].push_back({u,ww});
	}
	auto x = get_distance(E,0),y = get_distance(E,N-1);
	long long ans = x[N-1];
	
	
	LineContainer cht;
	rep(i,N){
		if(x[i]==Inf64)continue;
		cht.add(-w[i],-x[i]);
	}
	if(!cht.empty()){
	rep(i,N){
		if(y[i]==Inf64)continue;
		ans = min(ans,-cht.query(w[i]) + y[i]);
	}
	}
	cout<<ans<<endl;
	
	return 0;
}

yukicoder

結果

テストケース

ソースコード