Submission #72266619

---

Source Code Expand

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cmath>
#include <bitset>
#include <numeric>
using namespace std;
#define ll long long
#define vi vector<int>
#define vs vector<string>
#define vll vector<long long>
#define vii vector<pair<int, int>>
#define vllvector vector<pair<ll, ll>>
#define vb vector<bool>
#define yes cout << "Yes" << endl;
#define no cout << "No" << endl;
#define rep(i, c, n) for (int i = c; i < n; i++)
const long long llINF = 1LL << 60;
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true;} return false;}
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true;} return false;}
//! int型でinfを使うならINT_MAX使う

ll solve(ll n) {
        ll cur_total = 0;
    while(n > 0) {
        ll d = n % 10;
        cur_total += d * d;
        n /= 10;
    }
    return cur_total;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    ll n;
    cin >> n;
    while (n!=1 && n!=4) {
        n = solve(n);
    }

    if(n == 1) {
        yes
    } else {
        no
    }
}

Submission Info

Judge Result