#![allow(non_snake_case)] #[cfg(not(local))]macro_rules! eprintln{($($_:tt)*)=>{}} const MAX_TURN:usize=N*2; struct BUF{ base:i64, dist:Vec, prev:Vec>, que:BinaryHeap>>, que0:BinaryHeap>> } impl BUF{ fn new()->BUF{ BUF{ base:i64::MAX, dist:vec![i64::MAX;N2], prev:vec![vec![!0;MAX_TURN];N2], que:BinaryHeap::new(), que0:BinaryHeap::new() } } fn reset(&mut self){ self.base-=100000; self.que.clear(); } } // todo // pinch_penalty fn score(HP:i64,juwels:usize,key:bool)->f64{ const KEY_BONUS0:f64=3.; const KEY_BONUS1:f64=30.; juwels as f64 +if key{ let r=(1500-HP) as f64/1500.; KEY_BONUS0*(KEY_BONUS1/KEY_BONUS0).powf(r) } else { 0. } } #[derive(Clone)] struct Node{ hist:List, seen:Vec, pos:usize, turn:usize, HP:i64, juwels:usize, } impl Node{ fn new(input:&In)->Node{ Node{ hist:Cons(input.start,Rc::new(Nil)), seen:vec![0;(input.I>>3)+1], pos:input.start, turn:0, HP:H, juwels:0, } } fn get(&self,id:usize)->bool{ let (a,b)=(id>>3,id&7); self.seen[a]>>b&1==1 } fn to_goal(&self,input:&In,best:&mut List,best_score:&mut usize,BF:&mut BUF){ assert!(self.get(0)); let old=BF.base; BF.reset(); BF.que0.push(Reverse(S((BF.base,0),(self.pos,0)))); BF.dist[self.pos]=BF.base; BF.prev[self.pos][0]=!0; let th=BF.base+self.HP; while let Some(Reverse(S((cost,juwels),(pos,t))))=BF.que0.pop(){ let turn=(self.turn+t)%60; let nturn=(turn+1)%60; let nt=t+1; for dd in DX{ let np=pos+dd; let cell=input.grid[np]; let nc=input.damage[np][nturn]+cost; if cell==Goal{ let diff=cost-BF.base; if dd==0 || self.HP<=diff{ continue; } let mut route=vec![np]; let mut pos=(pos,t); loop{ route.push(pos.0); pos=(BF.prev[pos.0][pos.1],pos.1-1); if pos.0==!0{ break; } } route.pop(); route.reverse(); let juwels=self.juwels-juwels as usize; assert!(0nt{ if old>BF.dist[np] && dd!=0{ continue; } BF.dist[np]=old-1; if (BF.dist[np]>nc || dd==0) && th>nc{ BF.dist[np]=BF.dist[np].min(nc); BF.que0.push(Reverse(S((nc,juwels-matches!(cell,Juwel(id) if !self.get(id)) as i64),(np,nt)))); BF.prev[np][nt]=pos; } } } } } fn get_routes(&self,input:&In,BF:&mut BUF)->Vec<(Vec,i64,Cell)>{ // todo const C:usize=3; let R:i64=70.min(self.HP+0); let old=BF.base; BF.reset(); let mut ret=vec![]; BF.que.push(Reverse(S(BF.base,(self.pos,0)))); BF.dist[self.pos]=BF.base; BF.prev[self.pos][0]=!0; let th=BF.base+R; while let Some(Reverse(S(cost,(pos,t))))=BF.que.pop(){ let turn=(self.turn+t)%60; let nturn=(turn+1)%60; let nt=t+1; for dd in DX{ let np=pos+dd; let cell=input.grid[np]; let nc=input.damage[np][nturn]+cost; if cell==Goal && self.get(0) || matches!(cell,Juwel(id) if !self.get(id)){ if dd==0{ continue; } let nc=if cell==Goal{cost}else{nc}; let mut route=vec![np]; let mut pos=(pos,t); loop{ route.push(pos.0); pos=(BF.prev[pos.0][pos.1],pos.1-1); if pos.0==!0{ break; } } route.pop(); route.reverse(); ret.push((route,nc-BF.base,cell)); if ret.len()>=C{ return ret; } } else if cell.is_through() && 40>nt{ // todo if old>BF.dist[np] && dd!=0{ continue; } BF.dist[np]=old-1; if (BF.dist[np]>nc || dd==0) && th>nc{ BF.dist[np]=BF.dist[np].min(nc); BF.que.push(Reverse(S(nc,(np,nt)))); BF.prev[np][nt]=pos; } } } } ret } fn to_cand(&self,input:&In,id:(usize,usize),route:Vec,diff:i64)->Option{ let HP=self.HP-diff; let pos=*route.last().unwrap(); let key=(self.seen[0]&1==1)|(input.id[pos]==0); let cost=input.cost(pos,key); if HP<=0 || cost>=HP as f64+0.{ // todo return None; } let juwels=self.juwels+1; Some( Cand{ parent:id, HP, juwels,pos, turn:self.turn+route.len(), score:score(HP,juwels,key), op:route } ) } fn append_cands(&self,input:&In,id:(usize,usize),cands:&mut Vec>,best_score:&mut usize,best:&mut List,BF:&mut BUF){ let routes=self.get_routes(input,BF); for (route,diff,cell) in routes{ if let Some(cand)=self.to_cand(input,id,route,diff){ if cell==Goal{ if *best_score0); *best_score=cand.juwels; *best=self.hist.append(&cand.op); } } else{ cands[cand.HP as usize].push(cand); } } } } } #[derive(Clone)] struct Cand{ parent:(usize,usize), op:Vec, pos:usize, turn:usize, HP:i64, juwels:usize, score:f64, } impl Cand{ fn to_node(&self,input:&In,parent:&Node)->Node{ let mut seen=parent.seen.clone(); if let Juwel(id)=input.grid[self.pos]{ seen[id>>3]|=1<<(id&7); } Node{ seen, hist:parent.hist.append(&self.op), pos:self.pos, turn:self.turn, HP:self.HP, juwels:self.juwels, } } } struct Beam{ nodes:Vec>, best_score:usize, best:List } impl Beam{ fn new(node:Node)->Beam{ let mut nodes=vec![vec![];H as usize+1]; nodes[H as usize].push(node); Beam{ nodes, best_score:0, best:Nil } } fn enum_cands(&mut self,input:&In,HP:usize,cands:&mut Vec>,BF:&mut BUF){ for (i,node) in self.nodes[HP].iter().enumerate(){ node.append_cands(input,(HP,i),cands,&mut self.best_score,&mut self.best,BF); } } fn update<'a,I:Iterator>(&mut self,input:&In,HP:usize,it:I){ assert!(self.nodes[HP].is_empty()); for cand in it{ let (i,j)=cand.parent; let node=cand.to_node(input,&self.nodes[i][j]); self.nodes[HP].push(node); } } } fn solve(input:&In)->Vec{ // let mut M=100; // let mut BF=BUF::new(); // let mut beam=Beam::new(Node::new(input)); // let mut cands=vec![vec![];H as usize+1]; // let mut cnt=vec![0;input.I]; // let mut prev=H as usize; // let mut time0=get_time(); // for i in (1..=H as usize).rev(){ // if H as usize-1000>i && i&15==0 || i==112 || i==80 || i==48 || i==16 || i==8 || i==4 || i==2{ // let time1=get_time(); // if 2.6<=time1{ // M=70; // } // let mut rate=(prev-i) as f64*(2.6-time1)/i as f64/(time1-time0); // if rate>=1.{ // rate=rate.cbrt(); // } // M=((M as f64*rate) as usize).max(70).min((input.I as f64) as usize); // // eprintln!("# {M}"); // prev=i; // time0=time1; // } // beam.enum_cands(input,i,&mut cands,&mut BF); // if i==1{ // break; // } // cands[i-1].sort_unstable_by_key(|cand|Reverse(O(cand.score))); // cnt.fill(0); // let th=1; // todo // let it=cands[i-1].iter().filter(|cand|cnt[input.id[cand.pos]]=2.8{ // eprintln!("# {i}"); // break 'outer; // } // } // } // // しょうがないから自明解を出力 // if best_score==0{ let find_path=|start:usize,goal:usize|->(i64,Vec){ let mut prev=vec![!0;N2]; let mut que=BinaryHeap::new(); prev[start]=!1; que.push(Reverse(S((0,0),start))); let mut ans=0; 'outer: while let Some(Reverse(S((dist,juwels),pos)))=que.pop(){ for &dd in &DD[..4]{ let np=pos+dd; if np==goal{ prev[goal]=pos; ans=juwels; break 'outer; } let cell=input.grid[np]; if cell.is_through() && prev[np]==!0{ prev[np]=pos; que.push(Reverse(S((dist+1,juwels-matches!(cell,Juwel(_)) as i64),np))); } } } let mut route=vec![]; let mut pos=goal; loop{ route.push(pos); pos=prev[pos]; if pos==!1{ break; } } route.pop(); route.reverse(); (-ans,route) }; let mut route=vec![input.start]; let (score0,mut route0)=find_path(input.start,input.key); route.append(&mut route0); let (score1,mut route1)=find_path(input.key,input.goal); route.append(&mut route1); eprintln!("score = {}",score0+score1); return route; // } // let score=beam.best_score.saturating_sub(2); // assert!(score!=0); // eprintln!("score = {}",score); // best.to_vec() } fn simulate(input:&In,route:&[usize]){ let mut HP=H; for (i,p) in route.iter().cloned().enumerate(){ if i==0 || i==route.len()-1{ continue; } HP-=input.damage[p][i%60]; } assert!(HP>0); eprintln!("HP = {HP}"); } fn main(){ get_time(); let input=In::input(); eprintln!("I = {}",input.I); let ans=solve(&input); #[cfg(local)]{ simulate(&input,&ans); } for w in ans.windows(2){ let id=(0..5).find(|i|DX[*i]==w[1]-w[0]).unwrap(); println!("{}",DC[id]); } eprintln!("time = {:.2}",get_time()); } /////////////////////////////////// #[allow(unused)] struct Scanner{ stack:std::str::SplitAsciiWhitespace<'static> } #[allow(unused)] impl Scanner{ fn new()->Self{ use std::io::Read; let mut tmp=String::new(); std::io::stdin().read_to_string(&mut tmp).unwrap(); Self{stack:Box::leak(tmp.into_boxed_str()).split_ascii_whitespace()} } fn read(&mut self)->T{ self.stack.next().unwrap().parse::().unwrap_or_else(|_|panic!("parse error {}",std::any::type_name::())) } } macro_rules! input{ ($scan:ident, $($rest:tt)*)=>{ input_inner!{$scan,$($rest)*} }; } macro_rules! input_inner{ ($scan:ident $(,)?)=>{}; ($scan:ident, $name:ident:$t:tt $($rest:tt)*)=>{ let $name=read_value!($scan,$t); input_inner!{$scan $($rest)*} }; } macro_rules! read_value{ ($scan:ident, ($($t:tt),*))=>{ ($(read_value!($scan, $t)),*) }; ($scan:ident, [$t:tt;$len:expr])=>{ (0..$len).map(|_|read_value!($scan,$t)).collect::>() }; ($scan:ident, Chars)=>{ read_value!($scan,String).chars().collect::>() }; ($scan:ident, Usize1)=>{ read_value!($scan,usize)-1 }; ($scan:ident, $t:ty)=>{ $scan.read::<$t>() }; } use std::collections::*; use std::cmp::Reverse; use std::f64::INFINITY; #[derive(Clone,Copy,PartialEq,Eq,Debug)] enum Cell{ Empty, Wall, Goal, Juwel(usize), Fire, Enemy } use Cell::*; impl Cell{ fn new(c:char,at:&mut usize)->Cell{ match c{ '.'|'S'=>Empty, '#'=>Wall, 'G'=>Goal, 'K'=>Juwel(0), 'J'=>{*at+=1; Juwel(*at)}, 'F'=>Fire, 'E'=>Enemy, _=>panic!(), } } fn is_through(&self)->bool{ !matches!(*self,Wall|Goal|Enemy) } fn is_block(&self)->bool{ matches!(self,Wall|Enemy) } } const N:usize=60; const N2:usize=(N+2).pow(2); const H:i64=1500; fn get_cost(grid:&[Cell],costs:&[f64],start:usize)->Vec{ let mut que=BinaryHeap::new(); que.push(Reverse(S(0.,start))); let mut ret=vec![INFINITY;N2]; ret[start]=0.; while let Some(Reverse(S(cost,pos)))=que.pop(){ for dd in DD{ let np=pos+dd; if ret[np].is_infinite(){ let nc=if grid[np].is_through(){ cost+costs[np] } else{ continue; }; ret[np]=nc; que.push(Reverse(S(nc,np))); } } } ret } struct In{ I:usize, grid:Vec, id:Vec, start:usize, key:usize, goal:usize, cost_key:Vec, cost_goal:Vec, damage:Vec>, } impl In{ fn input()->In{ let mut scan=Scanner::new(); input!{ scan, n:usize, D:i64, h:i64, igrid:[Chars;n], M:usize, ienemy:[(usize,usize,usize);M] } assert_eq!((n,h),(N,H)); let mut grid=vec![Wall;N2]; let mut at=0; let mut start=!0; let mut key=!0; let mut goal=!0; for i in 1..=N{ for j in 1..=N{ let pos=to(i,j); let c=igrid[i-1][j-1]; grid[pos]=Cell::new(c,&mut at); if c=='S'{ start=pos; } else if c=='K'{ key=pos; } else if c=='G'{ goal=pos; } } } assert!(start!=!0 && key!=!0 && goal!=!0); let mut enemy=vec![!0;M]; let mut costf=vec![0.;N2]; let mut reach=vec![vec![];N2]; for (id,&(i,j,d)) in ienemy.iter().enumerate(){ enemy[id]=d; let pos=to(i+1,j+1); let addf=D as f64/d as f64+1.; for dd in DD{ let mut np=pos+dd; while !grid[np].is_block(){ costf[np]+=addf; reach[np].push(id); np+=dd; } } } let mut damage=vec![vec![0;60];N2]; for i in 1..=N{ for j in 1..=N{ for k in 0..60{ let cost=reach[to(i,j)].iter().map(|id|(k%enemy[*id]==0) as i64).sum::()*D+1; damage[to(i,j)][k]=cost; } } } let mut I=0; let mut id=vec![!0;N2]; for i in 1..=N{ for j in 1..=N{ if let Juwel(idx)=grid[to(i,j)]{ id[to(i,j)]=idx; I+=1; } } } In{ cost_key:get_cost(&grid,&costf,key), cost_goal:get_cost(&grid,&costf,goal), grid,start,key,goal, damage,I,id, } } fn cost(&self,pos:usize,key:bool)->f64{ if key{ self.cost_goal[pos] } else{ self.cost_goal[self.key]+self.cost_key[pos] } } } fn to(i:usize,j:usize)->usize{ i*(N+2)+j } const DD:[usize;4]=[!0,(N+2).wrapping_neg(),1,N+2]; const DX:[usize;5]=[!0,(N+2).wrapping_neg(),1,N+2,0]; const DC:[&'static str;5]=["M L","M U","M R","M D","S"]; #[derive(PartialEq,PartialOrd)] struct O(T); impl Eq for O{} impl Ord for O{ fn cmp(&self,a:&O)->std::cmp::Ordering{ self.0.partial_cmp(&a.0).unwrap() } } struct S(T,U); impl PartialEq for S{ fn eq(&self,a:&S)->bool{ self.0.eq(&a.0) } } impl PartialOrd for S{ fn partial_cmp(&self,a:&S)->Option{ self.0.partial_cmp(&a.0) } } impl Eq for S{} impl Ord for S{ fn cmp(&self,a:&S)->std::cmp::Ordering{ self.partial_cmp(a).unwrap() } } use std::rc::Rc; #[derive(Clone)] enum List{ Cons(T,Rc>), Nil } use List::*; impl List{ fn to_vec(&self)->Vec{ let mut ret=vec![]; let mut ptr=self.clone(); while let Cons(v,prev)=ptr{ ret.push(v.clone()); ptr=(*prev).clone(); } ret.reverse(); ret } fn append(&self,new:&[T])->List{ let mut ret=self.clone(); for n in new{ ret=Cons(n.clone(),Rc::new(ret)); } ret } } fn get_time()->f64{ static mut START:f64=-1.; let time=std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs_f64(); unsafe{ if START<0.{ START=time; } #[cfg(local)]{ (time-START)*1.5 } #[cfg(not(local))]{ time-START } } }