結果

問題 No.848 なかよし旅行
ユーザー threepipes_sthreepipes_s
提出日時 2019-07-05 23:42:32
言語 Java
(openjdk 23)
結果
AC  
実行時間 848 ms / 2,000 ms
コード長 4,831 bytes
コンパイル時間 2,839 ms
コンパイル使用メモリ 82,236 KB
実行使用メモリ 70,400 KB
最終ジャッジ日時 2024-11-08 00:56:12
合計ジャッジ時間 13,646 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.Closeable;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.lang.reflect.Array;
import java.util.*;

public class Main implements Runnable {
    static ContestScanner in;
    static Writer out;
    public static void main(String[] args) {
        new Thread(null, new Main(), "", 16 * 1024 * 1024).start();
    }

    public void run() {
        Main main = new Main();
        try {
            in = new ContestScanner();
            out = new Writer();
            main.solve();
            out.close();
            in.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    void solve() throws IOException {
        n = in.nextInt();
        m = in.nextInt();
        int p = in.nextInt() - 1;
        int q = in.nextInt() - 1;
        long t = in.nextLong();
        node = new List[n];
        for (int i = 0; i < n; i++) {
            node[i] = new ArrayList<>();
        }
        for (int i = 0; i < m; i++) {
            int a = in.nextInt() - 1;
            int b = in.nextInt() - 1;
            long c = in.nextLong();
            node[a].add(new Pos(b, c));
            node[b].add(new Pos(a, c));
        }
        long[] costS = getCost(0);
        long[] costP = getCost(p);
        long[] costQ = getCost(q);

        long max = -1;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                long cost = (costS[i] + costS[j] + Math.max(costP[i] + costP[j], costQ[i] + costQ[j]));
                if (cost > t) continue;
                max = Math.max(max, costS[i] + costS[j] + (t - cost));
                if (costS[i] + costP[i] + costQ[i] + costS[j] + costP[j] + costQ[j] <= t) {
                    max = t;
                }
            }
        }
        System.out.println(max);
    }

    int n, m;
    List<Pos>[] node;
    final long inf = Integer.MAX_VALUE * 2;
    long[] getCost(int s) {
        long[] costs = new long[n];
        boolean[] used = new boolean[n];
        Arrays.fill(costs, inf);
        Queue<Pos> pq = new PriorityQueue<>();
        pq.add(new Pos(s, 0));
        while (!pq.isEmpty()) {
            Pos p = pq.poll();
            if (used[p.p]) {
                continue;
            }
            used[p.p] = true;
            costs[p.p] = p.t;
            for (Pos e: node[p.p]) {
                pq.add(new Pos(e.p, p.t + e.t));
            }
        }
        return costs;
    }
}

class Pos implements Comparable<Pos> {
    long t;
    int p;
    Pos(int p, long t) {
        this.p = p;
        this.t = t;
    }

    @Override
    public int compareTo(Pos o) {
        return Long.compare(t, o.t);
    }
}


@SuppressWarnings("serial")
class MultiSet<T> extends HashMap<T, Integer>{
    @Override public Integer get(Object key){return containsKey(key)?super.get(key):0;}
    public void add(T key,int v){put(key,get(key)+v);}
    public void add(T key){put(key,get(key)+1);}
    public void sub(T key){final int v=get(key);if(v==1)remove(key);else put(key,v-1);}
    public MultiSet<T> merge(MultiSet<T> set)
    {MultiSet<T>s,l;if(this.size()<set.size()){s=this;l=set;}else{s=set;l=this;}
        for(Entry<T,Integer>e:s.entrySet())l.add(e.getKey(),e.getValue());return l;}
}

class Writer extends PrintWriter{
    public Writer(String filename)throws IOException
    {super(new BufferedWriter(new FileWriter(filename)));}
    public Writer()throws IOException{super(System.out);}
}
class ContestScanner implements Closeable{
    private BufferedReader in;private int c=-2;
    public ContestScanner()throws IOException
    {in=new BufferedReader(new InputStreamReader(System.in));}
    public ContestScanner(String filename)throws IOException
    {in=new BufferedReader(new InputStreamReader(new FileInputStream(filename)));}
    public String nextToken()throws IOException {
        StringBuilder sb=new StringBuilder();
        while((c=in.read())!=-1&&Character.isWhitespace(c));
        while(c!=-1&&!Character.isWhitespace(c)){sb.append((char)c);c=in.read();}
        return sb.toString();
    }
    public String readLine()throws IOException{
        StringBuilder sb=new StringBuilder();if(c==-2)c=in.read();
        while(c!=-1&&c!='\n'&&c!='\r'){sb.append((char)c);c=in.read();}
        return sb.toString();
    }
    public long nextLong()throws IOException,NumberFormatException
    {return Long.parseLong(nextToken());}
    public int nextInt()throws NumberFormatException,IOException
    {return(int)nextLong();}
    public double nextDouble()throws NumberFormatException,IOException
    {return Double.parseDouble(nextToken());}
    public void close() throws IOException {in.close();}
}
0