题目:给定一个只包含数字的字符串,复原它并返回所有可能的 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;
}