百度17年暑期实习生笔试题 - 单词接龙

问题

拉姆刚开始学习英文单词，对单词排序很感兴趣。
如果给拉姆一组单词，他能够迅速确定是否可以将这些单词排列在一个列表中，使得该列表中任何单词的首字母与前一单词的尾字母相同。
你能编写一个程序来帮助拉姆进行判断吗？

---

输入描述

输入包含多组测试数据。
对于每组测试数据，第一行为一个正整数n，代表有n个单词。
然后有n个字符串，代表n个单词。
保证：2<=n<=200,每个单词长度大于1且小于等于10,且所有单词都是由小写字母组成。

---

输出描述

对于每组数据，输出”Yes”或”No”

---

Java Code

public class WordSolitaire {    public static void main(String[] args) {        Scanner scan = new Scanner(System.in);        while(scan.hasNext()) {            int n = scan.nextInt();            String[] words = new String[n];            for (int i = 0; i < n; i++) {                words[i] = scan.next();            }            System.out.println(solve(n, words));        }        scan.close();    }       private static String solve(int n, String[] words) {        //用两个数组分别记录所有单词的首字母和尾字母，及相应字母重复出现的次数        int[] heads = new int[26];        int[] tails = new int[26];        while(--n >= 0) {            String s = words[n];            heads[s.charAt(0) - 97]++;//数组下标对应26个小写字母，元素值代表该字母出现在单词首位的次数            tails[s.charAt(s.length() - 1) - 97]++;        }        int count = 0;        for(int j = 0; j < 26; ++j) {            int diff = Math.abs(heads[j] - tails[j]);            if(diff > 1)                //如果当前字母作为单词的首字母和尾字母的次数之差超过1，                //则不能构成单词接龙                return "No";            else if(diff == 1) {                //如果当前字母作为单词的首字母和尾字母的次数之差等于1，                //但在此之前已经出现过满足这种情况的其他2个字母，则也不能构成单词接龙                if(++count > 2)                    return "No";            }        }        return "Yes";    }}

---

更新（2016.06.12）

import java.util.Scanner;public class WordSolitaire {    public static void main(String[] args) {        Scanner scan = new Scanner(System.in);        while(scan.hasNext()) {            int n = scan.nextInt();            String[] words = new String[n];            for (int i = 0; i < n; i++)                words[i] = scan.next();            System.out.println(solve(n, words));        }        scan.close();    }       private static String solve(int n, String[] words) {        //用两个数组分别记录所有单词的首字母和尾字母，及相应字母重复出现的次数        int[] heads = new int[26];        int[] tails = new int[26];        boolean flag = false;//标志位        while(--n >= 0) {            String s = words[n];            char head = s.charAt(0);            char tail = s.charAt(s.length() - 1);            //如果单词的首尾字母不同，分别记录之，否则忽略之            if(head != tail) {                heads[head - 97]++;//数组下标对应26个小写字母，元素值代表该字母出现在单词首位的次数                tails[tail - 97]++;                flag = true;            }        }        int count = 0;//记录作为单词的首字母和尾字母的次数之差等于1的字母个数        for(int j = 0; j < 26; ++j) {            int diff = Math.abs(heads[j] - tails[j]);            if(diff > 1)                //如果当前字母作为单词的首字母和尾字母的次数之差超过1，                //则不能构成单词接龙                return "No";            else if(diff == 1) {                //如果当前字母作为单词的首字母和尾字母的次数之差等于1，                //但在此之前已经出现过满足这种情况的其他2个字母，则也不能构成单词接龙                if(++count > 2)                    return "No";            }        }        //如果字母不总是成对地出现在同一单词的首位和末位，则可以构成单词接龙        return flag ? "Yes" : "No";    }}

---

说明

• 首先感谢@hellobug123 指出原版代码中的错漏之处。现已修改代码，增加对特殊情况的处理，如果还有错误之处，欢迎大家指出。

• 所给代码的时间复杂度为max(26, n)

---

更新（2016.06.14）

• 本文所给的解法仍然有漏洞，正确的解法请参考这里。