题解 | 魔法棒

import sys

T=int(input())
for i in range(T):
    x=int(input())
    #1.y=x+(k^2-1)
    # k=1 时，增量 = 0（相当于不操作）；
    # k=2 时，增量 = 4-1=3（最小非零增量）；
    # k=3 时，增量 = 9-1=8；
    # k=4 时，增量 = 16-1=15；
    # 以此类推（增量随 k 增大而增大）。
    
    # 2.因为最小有效增量为3，若存在3个连续的有效值，则可覆盖所有后续数

    # 3.[15,16,17]为连续有效数
    # 15=1->(1+3^2-1)=9->(9+2^2-1)=12->(12+2^2-1)=15
    # 16=1->(1+4^2-1)=16
    # 17=1->(1+3^2-1)=9->(9+3^2-1)=17
    
    # 所以当大于15时都可取到，只需分析小于15的数即可
    if x in [2,3,5,6,8,11,14]:
        print("No")
    else:
        print("Yes")