华为机试---投篮游戏



题目描述

有一个投篮游戏。球场有p个篮筐，编号为0，1...，p-1。每个篮筐下有个袋子，每个袋子最多装一个篮球。有n个篮球，每个球编号x_i 。规则是将数字为x_i 的篮球投到x_i 除p的余数为编号的袋里。若袋里已有篮球则球弹出游戏结束输出i，否则重复至所有球都投完。输出-1。问游戏最终的输出是什么？

输入描述:

第一行两个整数p,n（2≤p,n≤300)。p为篮筐数，n为篮球数。接着n行为篮球上的数字xi(0≤xi≤1e9)

输出描述:

输出游戏的结果

输入例子:

10 5
0
21
53
41
53

输出例子:

4

import java.util.Scanner;
 
public class Main {
 public static void main(String[] args){
  Scanner scan = new Scanner(System.in);
  while(scan.hasNext()){
   int p = scan.nextInt();
   int n = scan.nextInt();
   int[] ball_num = new int[n];
   for(int i = 0 ; i < n ; i++){
    ball_num[i] = scan.nextInt();
   }
   System.out.println(getFinalOutput(ball_num , p , n));
  }   
  scan.close();
    }
 private static int getFinalOutput(int[] ball_num, int p, int n) {
  int data = 0;
  int index = 0;//计算篮球应该保存在编号为index的篮筐下
  boolean[] flag = new boolean[p];
  int i = 0;
  for(; i < n ; i++){
   index = ball_num[i] % p;
   //编号为i的篮筐下已经有篮球，弹出篮球，返回i
   if(flag[index] == true){
    data = i + 1;
    break;
   }else{
    flag[index] = true;
   }
  }
  if(i == n){
   data = -1;
  }
  return data;
 }
 
}