글 읽기 - 어디가 틀렸을까요...

kchangwook 4년 전 0

아래처럼 작성했는데...

50%에서 자꾸 틀리네요...

어떤 반례가 있을까요?

import java.util.*;

public class Main {
	
	static int n, m, t, s, g, h;
	static List<int []> path[];
	static List<Integer> list;
	
	public static void main(String[] args) {

		Scanner input = new Scanner(System.in);
		int testCase = input.nextInt();
		for(int abcd = 0; abcd < testCase; abcd++) {
			
			n = input.nextInt();
			m = input.nextInt();
			t = input.nextInt();
			s = input.nextInt();
			g = input.nextInt();
			h = input.nextInt();
			
			int dist[] = new int[n + 1];
			path = new ArrayList[n + 1];
			list = new ArrayList<>();
			Queue<Integer> answer = new PriorityQueue<>();
			Queue<int []> que = new PriorityQueue<>((a, b) -> {
				if(a[1] < b[1]) return -1;
				else if(a[1] > b[1]) return 1;
				else return 0;
			}) ;
			
			for(int i = 0; i < m; i++) {
				int x = input.nextInt();
				int y = input.nextInt();
				int z = input.nextInt();
				if(path[x] == null) path[x] = new ArrayList<>();
				if(path[y] == null) path[y] = new ArrayList<>();
				
				int one[] = {y, z};
				int two[] = {x, z};
				
				path[x].add(one);
				path[y].add(two);
				
			}
			
			for(int i = 0; i < t; i++)
				list.add(input.nextInt());

			Arrays.fill(dist, Integer.MAX_VALUE);
			dist[s] = 0;
			int cde[] = {s, 0};
			que.add(cde);
			
			while(!que.isEmpty()) {
				
				int cur[] = que.poll();
				
				int x = cur[0];
				int y = cur[1];
				
				if(list.contains(x) && y == 1 && !answer.contains(x)) answer.add(x);
				
				for(int i = 0; i < path[x].size(); i++) {
					
					int ex[] = path[x].get(i);
					
					if(dist[ex[0]] >= dist[x] + ex[1]) {
						dist[ex[0]] = dist[x] + ex[1];
						int temp[] = new int[2];
						temp[0] = ex[0];
						if((x == g && ex[0] == h) || (x == h && ex[0] == g)) temp[1] = 1;
						else temp[1] = y;
						que.add(temp);
					}
					
				}
				
			}
			
			while(!answer.isEmpty()) {
				if(answer.size() == 1) System.out.println(answer.poll() + " ");
				else System.out.print(answer.poll() + " ");
			}
			
		}
		
	}// end of main
	
}// end of Main

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디가 틀렸을까요...

kchangwook 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말