leetcode 17. 电话号码的字母组合

电话号码的字母组合

 

给定一个仅包含数字 2-9 的字符串，返回所有它能表示的字母组合。答案可以按 任意顺序 返回。

给出数字到字母的映射如下（与电话按键相同）。注意 1 不对应任何字母。

示例 1：

输入：digits = "23"
输出：["ad","ae","af","bd","be","bf","cd","ce","cf"]

示例 2：

输入：digits = ""
输出：[]

示例 3：

输入：digits = "2"
输出：["a","b","c"]

提示：

• 0 <= digits.length <= 4
• digits[i] 是范围 ['2', '9'] 的一个数字。

 

思路：

先计算出来返回支付串的个数，也就是digits数组中对应数字所映射的字母个数的乘积；

然后为每个指针申请空间，因为digits.length <= 4，为了省事， 我直接申请了6个字节。

然后使用一个数组保留搜索到的字母，当digits为空时，表明搜索到结尾，将结果保留下来，然后回溯，继续搜索。

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

char *work[] = {    //按键上的字母
    "abc",
    "def",
    "ghi",
    "jkl",
    "mno",
    "pqrs",
    "tuv",
    "wxyz"
};
int count = 0;  

void dfs(char * digits, char *arr, char ** ans,int len, int index){
    if(*digits == '\0'){   //将本趟的结果保存下来
        arr[len] = '\0';
        strncpy(ans[count++], arr, len + 1);
        return ;
    }
    for(int i = 0; i < strlen(work[(digits[0] - '2')]); i++){
        arr[index] = work[(digits[0] - '2')][i];  
        dfs(digits + 1, arr, ans, len, index+1);   
    }
}

char ** letterCombinations(char * digits, int* returnSize){
    if(*digits == '\0'){    //digits 为空则直接返回
        * returnSize = 0;
        return NULL;
    }
        
    count = 0;
    *returnSize = 1;
    int len = strlen(digits);   
    for(int i = 0; digits[i]; i++){ //计算出返回的字符串个数
        *returnSize *= strlen(work[(digits[i] - '2')]);
    }
    //申请空间
    char **ans = (char **)malloc(sizeof(char *) * (*returnSize));
    for(int i = 0; i < *returnSize; i++){
        ans[i] = malloc(6); //为每个字符串申请空间
    }
    char arr[6] = {0};
    for(int i = 0; i < strlen(work[(digits[0] - '2')]); i++){
        arr[0] = work[(digits[0] - '2')][i];
        dfs(digits+1, arr, ans, len, 1);
    }
    return ans;
}