给你一个单词序列,判断其是否形成了一个有效的单词方块。
有效的单词方块是指此由单词序列组成的文字方块的 第 k 行 和 第 k 列 (0 ≤ k < max(行数, 列数)) 所显示的字符串完全相同。
注意:
给定的单词数大于等于 1 且不超过 500。
单词长度大于等于 1 且不超过 500。
每个单词只包含小写英文字母 a-z。
示例 1:
输入:
[
“abcd”,
“bnrt”,
“crmy”,
“dtye”
]
输出:
true
解释:
第 1 行和第 1 列都是 “abcd”。
第 2 行和第 2 列都是 “bnrt”。
第 3 行和第 3 列都是 “crmy”。
第 4 行和第 4 列都是 “dtye”。
因此,这是一个有效的单词方块。
示例 2:
输入:
[
“abcd”,
“bnrt”,
“crm”,
“dt”
]
输出:
true
解释:
第 1 行和第 1 列都是 “abcd”。
第 2 行和第 2 列都是 “bnrt”。
第 3 行和第 3 列都是 “crm”。
第 4 行和第 4 列都是 “dt”。
因此,这是一个有效的单词方块。
class Solution {
public boolean validWordSquare(List<String> words) {
/*
分别获取第i行的字符串和第j列的字符串,比较即可
时间O(MN)
*/
int len = words.size();
for(int i = 0; i < len; i++){
String rowStr = words.get(i); //第i行字符串
String colStr = getColumnStr(words, i);
if(!rowStr.equals(colStr)) return false;
}
return true;
}
public String getColumnStr(List<String>words, int j){
//取得第j列的字符串
int m = words.size();
StringBuffer sb = new StringBuffer();
for(int i = 0; i < m; i++){
String tmp = words.get(i);
if(j < tmp.length()){
sb.append(tmp.charAt(j));
}
}
return sb.toString();
}
}