LeetCode 202. 快樂數 (set記錄、出現循環節中的數則失敗)

快樂數

class Solution {
public:
    unordered_set<int> vis;
    bool isHappy(int n) {
        vis.insert(n);
        if(n==1){
            return 1;
        }
        while(1){
            long long sum = f(n);
            if(sum==1){
                return 1;
            }
            if(vis.count(sum)){
                return 0;
            }
            n = sum;
            vis.insert(n);
        }
        return 0;
    }
    long long f(int n){
        long long sum = 0;
        while(n){
            sum+= (long long)(n%10)*(n%10);
            n/=10;
        }
        return sum ;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章