题解 | 九倍平方数

#include <iostream>
#include <string>  // 必须包含string
using namespace std;

int main() {
    int t,a2,a3,a,i,j,flag;
    cin>>t;
    
    while(t--) 
    {
        flag=0;
        a2=0;a3=0;
        a=0;
        
        string s;
        cin>>s;
        
        for(int k=0;k<s.length();k++) 
        {
            int digit=s[k]-'0';
            a=(a+digit)%9;
            
            if(digit==2) a2++;
            else if(digit==3) a3++;
        }
        if(a==0) 
        {
            flag=1;
        }
        for(i=0;i<=a2&&!flag;i++) 
        {
            for(j=0;j<=a3&&!flag;j++) 
            {
                if(i==0&&j==0) continue;
                
                if((a+i*2+j*6)%9==0) 
                {
                    flag=1;
                }
            }
        }
        
        if(flag==1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    
    return 0;
}