Submission #72189802

Source Code Expand

from sys import stdin,stdout
# from os import _exit
# from bisect import bisect_left,bisect
# from heapq import heapify,heappop,heappush
# from sys import setrecursionlimit
# from collections import defaultdict,Counter,deque
# from itertools import permutations
# from math import gcd,ceil,sqrt,factorial
# setrecursionlimit(int(1e5))
input,print = stdin.readline,stdout.write

def f(n):
    ret = 0
    for i in str(n):
        ret+=pow(int(i),2)
    return ret

n = int(input())
s = set()
s.add(n)
for i in range(int(1e6)):
    n = f(n)
    if n in s:
        ans = "No"
    else:
        s.add(n)
        
if n==1:
    ans = "Yes"
print(f"{ans}\n")

Submission Info

Submission Time
Task B - Happy Number
User aviral
Language Python (PyPy 3.11-v7.3.20)
Score 200
Code Size 682 Byte
Status AC
Exec Time 140 ms
Memory 108496 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 200 / 200
Status
AC × 3
AC × 23
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt
Case Name Status Exec Time Memory
sample_01.txt AC 120 ms 108264 KiB
sample_02.txt AC 140 ms 108360 KiB
sample_03.txt AC 121 ms 108272 KiB
test_01.txt AC 113 ms 107992 KiB
test_02.txt AC 134 ms 108120 KiB
test_03.txt AC 121 ms 108116 KiB
test_04.txt AC 135 ms 108216 KiB
test_05.txt AC 121 ms 108092 KiB
test_06.txt AC 135 ms 108260 KiB
test_07.txt AC 122 ms 108092 KiB
test_08.txt AC 134 ms 108496 KiB
test_09.txt AC 121 ms 108092 KiB
test_10.txt AC 137 ms 108336 KiB
test_11.txt AC 122 ms 108204 KiB
test_12.txt AC 137 ms 108288 KiB
test_13.txt AC 122 ms 108144 KiB
test_14.txt AC 138 ms 108328 KiB
test_15.txt AC 122 ms 108140 KiB
test_16.txt AC 138 ms 108388 KiB
test_17.txt AC 123 ms 108064 KiB
test_18.txt AC 139 ms 108432 KiB
test_19.txt AC 121 ms 108140 KiB
test_20.txt AC 139 ms 108220 KiB