結果
問題 | No.1442 I-wate Shortest Path Problem |
ユーザー | kiyoshi0205 |
提出日時 | 2021-03-26 23:02:37 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 6,148 bytes |
コンパイル時間 | 3,048 ms |
コンパイル使用メモリ | 204,564 KB |
実行使用メモリ | 501,632 KB |
最終ジャッジ日時 | 2024-05-06 16:59:59 |
合計ジャッジ時間 | 9,952 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | AC | 5 ms
5,376 KB |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | AC | 131 ms
18,560 KB |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | TLE | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
ソースコード
#pragma GCC target("avx") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include<bits/stdc++.h> // #include<ext/pb_ds/assoc_container.hpp> // #include<ext/pb_ds/tree_policy.hpp> // #include<ext/pb_ds/tag_and_trait.hpp> // using namespace __gnu_pbds; // #include<boost/multiprecision/cpp_int.hpp> // namespace multiprecisioninteger = boost::multiprecision; // using cint=multiprecisioninteger::cpp_int; using namespace std; using ll=long long; using datas=pair<ll,ll>; using ddatas=pair<long double,long double>; using tdata=pair<ll,datas>; using vec=vector<ll>; using mat=vector<vec>; using pvec=vector<datas>; using pmat=vector<pvec>; // using llset=tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>; #define For(i,a,b) for(i=a;i<(ll)b;++i) #define bFor(i,b,a) for(i=b,--i;i>=(ll)a;--i) #define rep(i,N) For(i,0,N) #define rep1(i,N) For(i,1,N) #define brep(i,N) bFor(i,N,0) #define brep1(i,N) bFor(i,N,1) #define all(v) (v).begin(),(v).end() #define allr(v) (v).rbegin(),(v).rend() #define vsort(v) sort(all(v)) #define vrsort(v) sort(allr(v)) #define uniq(v) vsort(v);(v).erase(unique(all(v)),(v).end()) #define endl "\n" #define popcount __builtin_popcountll #define eb emplace_back #define print(x) cout<<x<<endl #define printyes print("Yes") #define printno print("No") #define printYES print("YES") #define printNO print("NO") #define output(v) do{bool f=0;for(auto outi:v){cout<<(f?" ":"")<<outi;f=1;}cout<<endl;}while(0) #define matoutput(v) do{for(auto outimat:v)output(outimat);}while(0) constexpr ll mod=1000000007; // constexpr ll mod=998244353; constexpr ll inf=1LL<<60; constexpr long double eps=1e-9; const long double PI=acosl(-1); template<class T,class E> ostream& operator<<(ostream& os,const pair<T,E>& p){return os<<"("<<p.first<<","<<p.second<<")";} template<class T> ostream& operator<<(ostream& os,const vector<T>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T> ostream& operator<<(ostream& os,const set<T>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T,class E> ostream& operator<<(ostream& os,const map<T,E>& v){ os<<"{";bool f=false; for(auto& x:v){if(f)os<<",";os<<x;f=true;} os<<"}"; return os; } template<class T> inline bool chmax(T& a,T b){bool x=a<b;if(x)a=b;return x;} template<class T> inline bool chmin(T& a,T b){bool x=a>b;if(x)a=b;return x;} #ifdef DEBUG void debugg(){cout<<endl;} template<class T,class... Args>void debugg(const T& x,const Args&... args){cout<<" "<<x;debugg(args...);} #define debug(...) cout<<__LINE__<<" ["<<#__VA_ARGS__<<"]:",debugg(__VA_ARGS__) #else #define debug(...) (void(0)) #endif void startupcpp(){ cin.tie(0); ios::sync_with_stdio(false); cout<<fixed<<setprecision(15); } ll modinv(ll a,ll m=mod) { ll b=m,u=1,v=0,t; while(b){ t=a/b; a-=t*b; swap(a,b); u-=t*v; swap(u,v); } return (u+m)%m; } ll moddevide(ll a,ll b){return (a*modinv(b))%mod;} vec modncrlistp,modncrlistm; ll modncr(ll n,ll r){ if(n<r)return 0; ll i,size=modncrlistp.size(); if(size<=n){ modncrlistp.resize(n+1); modncrlistm.resize(n+1); if(!size){ modncrlistp[0]=modncrlistm[0]=1; size++; } For(i,size,n+1)modncrlistp[i]=modncrlistp[i-1]*i%mod; modncrlistm[n]=modinv(modncrlistp[n]); for(i=n;i>size;--i)modncrlistm[i-1]=modncrlistm[i]*i%mod; } return modncrlistp[n]*modncrlistm[r]%mod*modncrlistm[n-r]%mod; } ll modpow(ll a,ll n,ll m=mod){ if(n<0)return 0; ll res=1; while(n>0){ if(n&1)res=res*a%m; a=a*a%m; n>>=1; } return res; } ll gcd(ll a,ll b){if(!b)return abs(a);return (a%b==0)?abs(b):gcd(b,a%b);} ll lcm(ll a,ll b){return a/gcd(a,b)*b;} struct HLD{ private: vector<vector<int>> g; int fdfs(int now,int p){ int d=0; par[now]=p; for(size_t i=0;i+1<g[now].size();++i){ if(g[now][i]==p)swap(g[now][i],g[now].back()); if(chmax(d,fdfs(g[now][i],now)))swap(g[now][0],g[now][i]); } g[now].pop_back(); return d+1; } void sdfs(int now){ in[now]=ETlist.size(); ETlist.emplace_back(now); for(size_t i=0;i<g[now].size();++i){ nxt[g[now][i]]=i?g[now][i]:nxt[now]; sdfs(g[now][i]); } out[now]=ETlist.size(); } public: // Heavy edge of i=[in[nxt[i]],in[i]] // sub tree of i=[in[i],out[i]) vector<int> par,nxt,in,out,ETlist; HLD(vector<vector<int>> G,int first=0):g(G),par(G.size()),nxt(G.size()),in(G.size()),out(G.size()){ ETlist.reserve(G.size()); init(first); }; HLD(int N):g(N),par(N),nxt(N),in(N),out(N){ETlist.reserve(N);}; void add_edge(int a,int b){ g[a].emplace_back(b); g[b].emplace_back(a); } void init(int first=0){ { int d=0;par[first]=-1; for(auto& x:g[first])if(chmax(d,fdfs(x,first)))swap(x,g[first][0]); } nxt[first]=first; sdfs(first); } int lca(int a,int b){ if(in[a]>in[b])swap(a,b); if(nxt[a]==nxt[b])return a; return lca(a,par[nxt[b]]); } }; int N,K,A,B,C; ll cost[1024][100002]; ll dfs(pmat& g,int bit,int now,int par=-1){ for(auto x:g[now]){ if(x.second==par)continue; chmin(cost[bit][x.second],cost[bit][now]+x.first); chmin(cost[bit][now],dfs(g,bit,x.second,now)+x.first); } return cost[bit][now]; } int main(){ startupcpp(); int i,j; cin>>N>>K; pmat g(N+1); HLD hld(N); rep1(i,N){ cin>>A>>B>>C; hld.add_edge(--A,--B); g[A].emplace_back(C,B); g[B].emplace_back(C,A); } hld.init(); { vector<int> fly[10],flycost(K); rep(i,K){ cin>>j>>flycost[i]; fly[i].reserve(j); fly[i].resize(j); for(auto& x:fly[i]){cin>>x;--x;} } rep(i,1<<K){ rep(j,N)cost[i][j]=inf; rep(j,K)if(i>>j&1)cost[i][N]+=flycost[j]; rep(j,K)if(i>>j&1)for(auto x:fly[j])cost[i][x]=0; dfs(g,i,0); } cost[0][0]=0; dfs(g,0,0); } int Q; cin>>Q; while(Q--){ cin>>A>>B; --A;--B; C=hld.lca(A,B); ll ans=cost[0][A]+cost[0][B]-cost[0][C]*2; rep(i,1<<K){ chmin(ans,cost[i][A]+cost[i][B]+cost[i][N]); } print(ans); } }