Java 最长单词

编写一个函数，输入一行字符，将此字符串中最长的单词输出。
　　输入仅一行，多个单词，每个单词间用一个空格隔开。单词仅由小写字母组成。所有单词的长度和不超过100000。如有多个最长单词，输出最先出现的。
样例输入
I am a student
样例输出
student
代码：

import java.util.Scanner;

public class Test {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s =sc.nextLine();
        String []arr = s.split(" ");
        int tmp = arr[0].length();
        int count = 0;
        for (int i = 1;i<arr.length;i++){
            if(arr[i].length() > tmp){
                tmp = arr[i].length();
                count = i;
            }
        }
        System.out.println(arr[count]);
    }
}