百度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）

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