利用判断链表是否有环的思想解决其他问题（leetcode141）

利用判断链表是否有环的思想解决其他问题（leetcode141）

昨天学习了一种方法判断链表是否有环，是利用两个指针一个，一个快指针一个慢指针来遍历链表，如果链表没有环那么肯定能遍历到尾节点。
但是如果有环的话，在某一时刻这两个指针一定会指向同一个结点。
代码如下：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
bool hasCycle(struct ListNode *head) {
    if (head == NULL || head->next == NULL) return false;
    struct ListNode *fast_node, *slow_node;
    fast_node = slow_node = head;
    while (fast_node->next->next != NULL) {
            fast_node = fast_node->next->next;//快指针走两步
            slow_node = slow_node->next;//慢指针走一步
        if (fast_node == slow_node) {//当某一时刻两个指针相等时，代表链表有环
            return true;
        }
        if (fast_node->next == NULL) return false;//当遍历到尾结点时，链表无环
    }
    return false;
}

但是这种思想不仅能解决判断链表是否有环的问题，也能解决类似的问题，比如leetcode202
问题是给一个非负数n，计算他是不是开心数。
开心数具有如下定义：对n的各个位的平方相加，得到新的n，在某一次可以使得n为1，那么这个数就是开心数。
一开始我的想法就是建立一个足够大的数组，把每一次n的值存进去，当得到新的n时，就把n和数组中的每一个数进行比较，如果数组中存在某一个数和新的n相等，那么就说明形成了一个环，n就不是开心数。
代码如下：

bool isHappy(int n) {
    if (n == 1) return true;
    int a[10000] = {0};//数组大小为10000，足够大
    a[0] = 1;//a[0]用来存储数组中元素的个数
    a[1] = n;//当n不是1的时候，存入数组中
    while (1) {
        int pre_n = n;
        n = 0;
        while (pre_n) {
            n = n + (pre_n % 10) * (pre_n % 10);
            pre_n /= 10;
        }
        if (n == 1) return true;
        for (int i = 1; i <= a[0]; i++) {
            if (a[i] == n) return false;
        }
        a[++a[0]] = n;
    }
}

但是上述的方法时间复杂度为O(n^2),如果可以使用我们之前的方法，设定两个数fast和slow，就可以更好的解决这个问题。
代码如下：

int func(int n) {//用来计算n的各个位平方之和
    int sum = 0;
    while (n) {
        sum = sum + (n % 10) * (n % 10);
        n /= 10;
    }
    return sum;
}
bool isHappy(int n) {
    if (n < 1) return false;
    int fast, slow;//设定两个数，fast和slow
    fast = slow = n;
    while (fast != 1 && slow != 1) {
        fast = func(func(fast));//fast走两步
        slow = func(slow);//slow走一步
        if (fast == 1 || slow == 1) return true; //先判断是否有数为1
        if (fast == slow) {//当两个数相等时说明有环，这个数就不是开心数
            return false;
        }
    }
    return true;
}