leetcode[93]复原ip地址

题目：给定一个只包含数字的字符串，复原它并返回所有可能的 IP 地址格式。

示例:

输入: "25525511135"
输出: ["255.255.11.135", "255.255.111.35"]

 

源码：

/* 判断是否为有效IP */
/* 长度为1-3, 没有前导0, 值在[0,255]之间 */
bool isValidIp(char *s, int len, int *buf, int idx) {
    if (len >= 2 && s[0] == '0') {
        return false;
    }
    if (len == 3 && s[0] == '0' && s[1] == '0') {
        return false;
    }
    int i;
    int tmp = 0;
    for (i = 0; i < len; i++) {
        tmp = tmp * 10 + (s[i] - '0');
    }
    if (tmp > 255) {
        return false;
    } else {
        buf[idx] = tmp;
        return true;
    }
}

void dfs(char *s, char **res, int *returnSize, int *buf, int idx) {
    if (idx == 4) {
        if (*s == '\0') { /* 符合条件, 结果输出 */
            sprintf(res[(*returnSize)++], "%d.%d.%d.%d", buf[0], buf[1], buf[2], buf[3]);
        }
        return;
    }
    int i;
    for (i = 1; i <= 3; i++) {
        if (i > strlen(s) || strlen(s) > ((4 - idx) * 3)) { /* 剪枝优化, 长度过长或过短时都是非法 */
            break;
        }
        if (isValidIp(s, i, buf, idx) == false) { /* 非有效IP跳过, 有效IP则存入buf[idx]中 */
            continue;
        }
        dfs(s + i, res, returnSize, buf, idx + 1); /* 回溯算法 */
    }
}
/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
char ** restoreIpAddresses(char * s, int* returnSize){
    if (s == NULL || strlen(s) == 0 || strlen(s) < 4 || strlen(s) > 12) { /* 异常判断 */
        *returnSize = 0;
        return NULL;
    }
    int len = strlen(s);
    int *buf = (int*)malloc(sizeof(int) * 4); /* 最多4个IP数字 */
    char **res = (char**)malloc(sizeof(char*) * 100);
    int i;
    for (i = 0; i < 100; i++) {
        res[i] = (char*)malloc(sizeof(char) * (len + 4)); /* 3个逗号加结束符 */
    }
    *returnSize = 0;
    dfs(s, res, returnSize, buf, 0);
    return res;
}