移动字母

题目链接:https://ac.nowcoder.com/acm/contest/6218/A

Description

给定一个只包含小写字母的字符串s，牛牛想将这个字符串中的所有’a’字母全部移动到字符串的末尾，而且保证其它字符的相对顺序不变。其中字符串s的长度<=1e6。

Solution

字符串长度较小，直接遍历一遍即可。定义一个新的空字符串str，和一个计数器num，遍历字符串s时，如果s[i]不是’a’，就将该字符加入字符串str，否则，num++，记录一下’a’出现的次数。最后在str后加上num个’a’。

Code_1

class Solution {
public:
    /**
     *
     * @param s string字符串
     * @return string字符串
     */
    string change(string s) {
        // write code here
        string str;
        int num=0;
        for(int i=0;i<s.length();i++){
            if(s[i]!='a') str+=s[i];
            else num++;
        }
        for(int i=0;i<num;i++) str+='a';
        return str;
    }
};

也可以用一下remove()函数

Code_2

class Solution {
public:
    /**
     * 
     * @param s string字符串 
     * @return string字符串
     */
    string change(string s) {
        // write code here
        int num = 0;
	    for (int i = 0; i < s.length(); i++) if (s[i] == 'a') num++;
	    remove(s.begin(), s.end(), 'a');
	    for (int i = 0; i < num; i++) s[s.length() - i - 1] = 'a';
        return s;
    }
};