算法修炼之路——【字符串】Leetcode 824 山羊拉丁文

题目描述

给定一个由空格分隔单词的句子s。每个单词只包含大写或小写字母。

我们要将句子转换为Goat Latin（一种类似于猪拉丁文-Pig Latin的虚构语言）。

山羊拉丁文的规则如下：

• 如果单词以元音开头（a, e, i, o, u），在单词后添加"ma"；如单词apple变为applema；
  如果单词以辅音字母开头，移出第一个字符并将它放到末尾，之后再添加"ma"; 如单词goat变为oatgma；
  根据单词在句子中的索引，在单词最后添加与索引相同数量的字母a，索引从1开始；例如，在第一个单词后添加"a",在第二个但此后添加"aa"，以此类推。

结果为将s转换为山羊拉丁文的字符串句子。

示例1：

输入： s = “I speak Goat Latin”
输出： “Imaa peaksmaaa oatGmaaaa atinLmaaaaa”

示例2：

输入： s = “The quick brown fox jumped over the lazy dog”
输出： “heTmaa uickqmaaa rownbmaaaa oxfmaaaa umpedjmaaaaaa overmaaaaaaa hetmaaaaaaaa azylmaaaaaaaaa ogdmaaaaaaaaaa”

说明：

1. s中仅包含大小写字母和空格。单词间有且仅有一个空格；
2. s长处于[1, 150]之间。

思路分析

题目难度为简单 ，这里并不需要特别复杂的思路，只需要将字符串按照空格提取单词，并判断首字母是否为元音，之后添加相应的ma与若干a即可。

这里我们为了简化元音判断，直接将a，e，i，o，u存入HashSet中，此时访问复杂度为O(1)；结果通过StringBuilder res存储，且每个单词判断之后加入空格" "，返回[0, res.length() - 1]即为所求。

解题代码

    public static String solution(String s) {
        if(s == null) return null;
        StringBuilder res = new StringBuilder();

        Character[] chars = {'a','e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U'};
        set.addAll(Arrays.asList(chars)); 
        
        int index = 1;
        for(String word : s.split(" ")){
            char firstC = word.charAt(0);
            if(!set.contains(firstC)){
                res.append(word.substring(1));
                res.append(firstC);
                res.append("ma");
            }
            for(int i = 0; i < index; i++)
                res.append("a");
            res.append(" ");
            
            index++;
        }
        return res.substring(0, res.length()-1);
    }

复杂度分析

时间复杂度： 不仅要遍历源字符，还要考虑增加的maaa...,故时间复杂度为O(n^2);
空间复杂度： 我们借用了StringBuiler存储输出，其包括输入字符以及新增的a字符，其空间复杂度为O(n^2);

Github源码

完整可运行文件请访问GitHub。