글 읽기 - 5670번 자바로 푸는데 틀렸습니다가 뜨네요..

jjy5349 3년 전 0

입력 예시들이랑 제가 생각하는 입력들은 모두 정상 출력되는데 어디가 틀렸는지 계속 틀렸습니다가 뜨네요...

제 눈에는 도저히 안보이는데 혹시 도와주실 수 있나요?

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		int n;
		String input = new String();
		ArrayList <String> word = new ArrayList<>(); 
		trie T = new trie();
		int sum=0;
		double avg;
		
		n = sc.nextInt();
		for(int i=0; i<n; i++) // 단어 입력과 노드 추가
		{
			input = sc.next();
			word.add(input);
			T.insert(input);
		} 
		
		for(int i=0; i<n; i++) // 각 입력 횟수의 합을 구하기 
			sum += T.find( T, word.get(i) );
		
		avg = (double)sum / (double)n;
		System.out.format("%.2f%n", avg);
		
	}
}

class trie_Node 
{
	Map<Character, trie_Node> chd_Node = new HashMap<>();
	boolean last_check = false;	
}

class trie
{
	trie_Node root = new trie_Node();
	
	void insert(String input)
	{
		trie_Node temp = this.root;
		
		for(int i=0; i<input.length(); i++)
		{
			if( temp.chd_Node.get( input.charAt(i)) == null )
			{
				temp.chd_Node.put( input.charAt(i), new trie_Node() );
				temp = temp.chd_Node.get( input.charAt(i) );
			}
			else
				temp = temp.chd_Node.get( input.charAt(i) );
		}
		temp.last_check = true;
	}
	
	
	int find(trie T, String input)
	{
		int count = 1;
		trie_Node temp = T.root.chd_Node.get( input.charAt(0) ); // 먼저 1번 입력했다고 가정하고 시작합니다.
		
		for(int i=1; i<input.length(); i++)
		{
			if(temp.chd_Node.size() >= 2)
				count++;
			
			else if( temp.chd_Node.size() == 1 && temp.last_check == true )
			{
				count++;
			}
			
			temp = temp.chd_Node.get( input.charAt(i) );
			
		}
		
		
		return count;
	}
}

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

문제

문제

출처

ICPC

5670번 자바로 푸는데 틀렸습니다가 뜨네요..

jjy5349 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말