认老乡-编程题有感

题目描述

大学的同学来自全国各地，对于远离家乡步入陌生大学校园的大一新生来说，碰到老乡是多么激动的一件事，于是大家都热衷于问身边的同学是否与自己同乡，来自新疆的小赛尤其热衷。但是大家都不告诉小赛他们来自哪里，只是说与谁同乡，从所给的信息中，你能告诉小赛有多少人确定是她的同乡吗？

输入

每个测试实例首先包括2个整数，N（1 <= N <= 1000），M(0 <= M <= N*(N-1)/2)，代表现有N个人（用1~N编号）和M组关系；

在接下来的M行里，每行包括2个整数，a，b，代表a跟b是同乡；

当N = 0，M = 0输入结束；

已知1表示小赛本人。

输出

对于每个测试实例，输出一个整数，代表确定是小赛同乡的人数。

样例输入

3 1

2 3

5 4

1 2

3 4

2 5

3 2

0 0

样例输出

0

4

做这道题目时候也是一波三折，题目本身不难，但是需要注意的点有很多

（1）数据结构的选择，主要是关于集合的选择上

（2）题目条件的注意

代码1

import java.util.Scanner;
import java.util.*;
public class Main {

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int M,N;	
		while(in.hasNext()){
			N=in.nextInt();
			M=in.nextInt();
			if(N==0 && M==0){
				break;
			}
			if(N<1 || N>1000 || M<0 || M>(N*(N-1)/2)){
				continue;
			}else{
				int[][] ralation = new int[1][2];
				HashMap
   
   
    
    
     
     > relationMap = new HashMap
     
     
      
      >();
				for (int i = 0; i < M; i++) {
					ralation[0][0]=in.nextInt();
					ralation[0][1]=in.nextInt();
					if(relationMap.get(ralation[0][0])==null){
						ArrayList
      
      
       
        list = new ArrayList
       
       
         (); list.add(ralation[0][1]); relationMap.put(ralation[0][0], list); }else{ relationMap.get(ralation[0][0]).add(ralation[0][1]); } if(relationMap.get(ralation[0][1])==null){ ArrayList 
        
          list = new ArrayList 
         
           (); list.add(ralation[0][0]); relationMap.put(ralation[0][1], list); }else{ relationMap.get(ralation[0][1]).add(ralation[0][0]); } } if(relationMap.get(1)==null){ System.out.println(0); }else{ HashSet 
          
            hasVisited = new HashSet 
           
             (); HashSet 
            
              notVisited = new HashSet 
             
               (); notVisited.addAll(relationMap.get(1)); while(notVisited.size()>0){ int num = notVisited.iterator().next(); notVisited.remove(num); hasVisited.add(num); if(relationMap.get(num)!=null){ notVisited.addAll(relationMap.get(num)); notVisited.removeAll(hasVisited); } } if(hasVisited.contains(1)){ System.out.println(hasVisited.size()-1); }else{ System.out.println(hasVisited.size()); } } } } } } 
              
             
            
           
          
         
       
      
      
     
     
    
    
   
   

代码2

import java.util.Scanner;
import java.util.*;
public class Main {

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int M,N;	
		while(in.hasNext()){
			N=in.nextInt();
			M=in.nextInt();
			if(N==0 && M==0){
				break;
			}
			if(N<1 || N>1000 || M<0 || M>(N*(N-1)/2)){
				continue;
			}else{
				int[][] ralation = new int[1][2];
				HashMap
   
   
    
    
     
     > relationMap = new HashMap
     
     
      
      >();
				for (int i = 0; i < M; i++) {
					ralation[0][0]=in.nextInt();
					ralation[0][1]=in.nextInt();
					if(relationMap.get(ralation[0][0])==null){
						ArrayList
      
      
       
        list = new ArrayList
       
       
         (); list.add(ralation[0][1]); relationMap.put(ralation[0][0], list); }else{ relationMap.get(ralation[0][0]).add(ralation[0][1]); } if(relationMap.get(ralation[0][1])==null){ ArrayList 
        
          list = new ArrayList 
         
           (); list.add(ralation[0][0]); relationMap.put(ralation[0][1], list); }else{ relationMap.get(ralation[0][1]).add(ralation[0][0]); } } if(relationMap.get(1)==null){ System.out.println(0); }else{ HashSet 
          
            hasVisited = new HashSet 
           
             (); ArrayList 
            
              notVisited = new ArrayList 
             
               (); notVisited.addAll(relationMap.get(1)); while(notVisited.size()>0){ int num = notVisited.remove(0); hasVisited.add(num); if(relationMap.get(num)!=null){ for (int i = 0; i < relationMap.get(num).size(); i++) { if(!hasVisited.contains(relationMap.get(num).get(i))){ notVisited.add(relationMap.get(num).get(i)); } } } } if(hasVisited.contains(1)){ System.out.println(hasVisited.size()-1); }else{ System.out.println(hasVisited.size()); } } } } } } 
              
             
            
           
          
         
       
      
      
     
     
    
    
   
   

代码1和代码2在后期数据处理的时候由于选用不同的数据结构，在效率上代码1明显高于代码2，这也是一个值得深思的问题，如何合理选用java提供的集合元素来满足程序功能的同时也达到高效的程序执行效果。代码2程序的执行时间在数据量大的时候会超时，代码1在这方面则比较有效。