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

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