結果
問題 | No.1094 木登り / Climbing tree |
ユーザー | S97323424 |
提出日時 | 2020-10-18 14:38:14 |
言語 | Lua (LuaJit 2.1.1696795921) |
結果 |
AC
|
実行時間 | 1,418 ms / 2,000 ms |
コード長 | 2,888 bytes |
コンパイル時間 | 335 ms |
コンパイル使用メモリ | 7,200 KB |
実行使用メモリ | 117,680 KB |
最終ジャッジ日時 | 2024-04-25 19:30:51 |
合計ジャッジ時間 | 28,211 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 1,121 ms
115,208 KB |
testcase_02 | AC | 298 ms
109,056 KB |
testcase_03 | AC | 199 ms
10,624 KB |
testcase_04 | AC | 510 ms
53,916 KB |
testcase_05 | AC | 883 ms
108,320 KB |
testcase_06 | AC | 631 ms
31,128 KB |
testcase_07 | AC | 1,174 ms
115,448 KB |
testcase_08 | AC | 1,139 ms
115,436 KB |
testcase_09 | AC | 1,164 ms
115,440 KB |
testcase_10 | AC | 1,131 ms
115,524 KB |
testcase_11 | AC | 1,322 ms
115,532 KB |
testcase_12 | AC | 1,312 ms
116,300 KB |
testcase_13 | AC | 1,348 ms
116,716 KB |
testcase_14 | AC | 1,418 ms
117,184 KB |
testcase_15 | AC | 433 ms
29,312 KB |
testcase_16 | AC | 708 ms
95,616 KB |
testcase_17 | AC | 574 ms
49,664 KB |
testcase_18 | AC | 504 ms
45,568 KB |
testcase_19 | AC | 654 ms
63,616 KB |
testcase_20 | AC | 1,412 ms
117,680 KB |
testcase_21 | AC | 579 ms
52,424 KB |
testcase_22 | AC | 1,161 ms
115,560 KB |
testcase_23 | AC | 1,165 ms
115,504 KB |
testcase_24 | AC | 1,124 ms
115,520 KB |
testcase_25 | AC | 1,132 ms
115,432 KB |
testcase_26 | AC | 1,110 ms
115,312 KB |
ソースコード
local function meld(node1,node2) if not node1 then return node2 elseif not node2 then return node1 elseif node1[1]<node2[1] then node2[4]=node1[3] node1[3]=node2 return node1 else node1[4]=node2[3] node2[3]=node1 return node2 end end local function pairing(node,stack) if node and node[4] then return pairing(node[4][4],rawset(meld(node,node[4]),4,stack)) else return stack,node end end local function meld_pairs(stack,node) if stack then return meld_pairs(stack[4],meld(stack,node)) else return node end end local PairingHeap={} PairingHeap.empty=function(self) return not self[1] end PairingHeap.push=function(self,key,value) self[1]=meld(self[1],{key,value,false,false}) end PairingHeap.top=function(self) return self[1][2] end PairingHeap.pop=function(self) self[1]=meld_pairs(pairing(self[1][3],false)) end PairingHeap.new=function() return setmetatable({false},{__index=PairingHeap}) end ---------- local INF=10^20 local function Dijkstra(graph,start) local dist={} local rank={0} local par={} for i=1,#graph do dist[i]=INF par[i]=start end local pq=PairingHeap:new() dist[start]=0 pq:push(0,{0,start}) while not pq:empty() do local p=pq:top() pq:pop() local from=p[2] if dist[from]<p[1] then goto continue end for to,cost in pairs(graph[from]) do if dist[to]>dist[from]+cost then dist[to]=dist[from]+cost rank[to]=rank[from]+1 par[to]=from pq:push(dist[to],{dist[to],to}) end end ::continue:: end return dist,rank,par end ---------- local n=io.read("*n") local tree={} for i=1,n do tree[i]={} end for i=1,n-1 do local a,b,c=io.read("*n","*n","*n") tree[a][b]=c tree[b][a]=c end local root=1 local dist local rank local par={} dist,rank,par[1]=Dijkstra(tree,root) local bit=require("bit") local k=1 while bit.lshift(1,k)<n do k=k+1 end for i=2,k do par[i]={} for j=1,n do par[i][j]=root end end for i=1,k-1 do for j=1,n do if par[i][j]~=root then par[i+1][j]=par[i][par[i][j]] end end end local q=io.read("*n") for i=1,q do local s,t=io.read("*n","*n") local length=dist[s]+dist[t] if rank[s]<rank[t] then s,t=t,s end for j=1,k do if bit.band(bit.rshift((rank[s]-rank[t]),(j-1)),1)>0 then s=par[j][s] end end if s~=t then for j=k,1,-1 do if par[j][s]~=par[j][t] then s=par[j][s] t=par[j][t] end end s=par[1][s] end io.write(length-2*dist[s].."\n") end