2011年2月15日火曜日

Aizu Online Judge 0117 A Reward a Carpenter

■0117 A Reward a Carpenter

(殿様から大工が受け取ったお金)-(柱の代金)-(出発町から山里までの総交通費)-(山里から出発町までの総交通費)が答え.後ろ2つは,それぞれDijkstraで求めれば良い.

import java.util.*;
import java.lang.*;
import java.math.*;
import java.io.*;

import static java.lang.Math.*;
import static java.util.Arrays.*;

public class Main{

 Scanner sc=new Scanner(System.in);

 int INF=1<<28;
 double EPS=1e-9;

 int n, m;

 void run(){
  sc.useDelimiter("[,]|(\\s)+");
  n=sc.nextInt();
  m=sc.nextInt();
  @SuppressWarnings("unchecked")
  LinkedList<E>[] es=new LinkedList[n];
  for(int i=0; i<n; i++){
   es[i]=new LinkedList<E>();
  }

  for(int i=0; i<m; i++){
   int u=sc.nextInt()-1;
   int v=sc.nextInt()-1;
   int cost1=sc.nextInt();
   int cost2=sc.nextInt();
   es[u].add(new E(v, cost1));
   es[v].add(new E(u, cost2));
  }

  int s=sc.nextInt()-1;
  int g=sc.nextInt()-1;
  int m1=sc.nextInt();
  int m2=sc.nextInt();

  int ans=m1-m2-(dijkstra(es, s, g)+dijkstra(es, g, s));
  println(""+ans);
 }

 int dijkstra(LinkedList<E>[] es, int s, int g){
  int n=es.length;
  int[] d=new int[n];
  PriorityQueue<P> que=new PriorityQueue<P>();

  Arrays.fill(d, INF);
  d[s]=0;
  que.offer(new P(s, 0));
  for(; !que.isEmpty();){
   P p=que.poll();
   if(d[p.v]<p.d){
    continue;
   }
   for(E e : es[p.v]){
    if(d[e.to]>d[p.v]+e.cost){
     d[e.to]=d[p.v]+e.cost;
     que.offer(new P(e.to, d[e.to]));
    }
   }
  }
  return d[g];
 }

 class E{
  int to, cost;

  E(int to, int cost){
   this.to=to;
   this.cost=cost;
  }
 }

 class P implements Comparable<P>{
  int v, d;

  P(int v, int d){
   this.v=v;
   this.d=d;
  }

  @Override
  public int compareTo(P p){
   return d-p.d;
  }

  @Override
  public String toString(){
   return v+","+d;
  }
 }

 void debug(Object... os){
  System.err.println(Arrays.deepToString(os));
 }

 void print(String s){
  System.out.print(s);
 }

 void println(String s){
  System.out.println(s);
 }

 public static void main(String[] args){
  // System.setOut(new PrintStream(new BufferedOutputStream(System.out)));
  new Main().run();
 }
}

0 件のコメント: