Java程序练习-生日相同

http://poj.grids.cn/practice/3238/
生日相同



时间限制: 1000ms内存限制: 65536kB

描述

在一个有180人的大班级中，存在两个人生日相同的概率非常大，现给出每个学生的学号，出生月日。试找出所有生日相同的学生。

输入

第一行为整数n，表示有n个学生，n<100。

此后每行包含一个字符串和两个整数，分别表示学生的学号（字符串长度小于10）和出生月(1<=m<=12)日(1<=d<=31)。

学号、月、日之间用一个空格分隔。

输出

对每组生日相同的学生，输出一行，

其中前两个数字表示月和日，后面跟着所有在当天出生的学生的学号，数字、学号之间都用一个空格分隔。

对所有的输出，要求按日期从前到后的顺序输出。

对生日相同的学号，按输入的顺序输出。

样例输入

5

00508192 3 2

00508153 4 5

00508172 3 2

00508023 4 5

00509122 4 5

样例输出

3 2 00508192 00508172

4 5 00508153 00508023 00509122

参考代码

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
public class Main {
public static void main(String[] args) throws Exception, IOException {
BufferedReader cin = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(cin.readLine());
Map<Integer,List<String>> map = new TreeMap<Integer,List<String>>();
while(n > 0){
String ds[] = cin.readLine().split(" ");
String id = ds[0];
int birth = Integer.parseInt(ds[1])*100 + Integer.parseInt(ds[2]);
List<String> list = new ArrayList<String>();
if(map.containsKey(birth)){
list = map.get(birth);
list.add(id);
map.put(birth, list);
}else{
list.add(id);
map.put(birth, list);
}
n --;
}
Set<Integer> set = map.keySet();
Iterator<Integer> it = set.iterator();
while(it.hasNext()){
Integer birth = it.next();
List<String> list = map.get(birth);
if(list.size() < 2)
continue;
System.out.print(birth / 100 +" "+ birth % 100);
for(int i = 0;i < list.size();++ i){
System.out.print(" "+list.get(i));
}
System.out.println();
}
}
}