leetcode-791. Custom string sorting

Given two strings order and s .order All the words are only Of , And it used to be sorted in some custom order .

Yes s Replace the characters of , Align it with the sorted  order  Match . More specifically , If in  order  The characters in x The character... Appears y Before , Then in the arranged string , x  It should also appear in y Before .

return Satisfying this property s Arbitrary arrangement of  .

Example 1:

Input : order = “cba”, s = “abcd”
Output : “cbad”
explain :
“a”、“b”、“c” It appears in order , therefore “a”、“b”、“c” The order of should be “c”、“b”、“a”.
because “d” Not in order , So it can be anywhere in the returned string .“dcba”、“cdba”、“cbda” It is also a valid output .
Example 2:

Input : order = “cbafg”, s = “abcd”
Output : “cbad”

Tips :

1 <= order.length <= 26
1 <= s.length <= 200
order  and  s  It's made up of lowercase letters
order  All characters in the are Different

java Code ：

class Solution {
    public String customSortString(String order, String s) {
        int[] count = new int[26];
        for (int i = 0; i < s.length(); i++) {
            count[s.charAt(i) - 'a']++;
        }

        StringBuilder sbd = new StringBuilder();
        for (char c : order.toCharArray()) {
            for (int i = 0; i < count[c - 'a']; i++) {
                sbd.append(c);
            }
            count[c - 'a'] = 0;
        }

        for (char c = 'a'; c <= 'z'; c++) {
            for (int i = 0; i < count[c - 'a']; i++) {
                sbd.append(c);
            }
        }
        return sbd.toString();

    }
}