洛谷 B3888 [语言月赛 202311] 画图练习

#include <iostream>
#include<bits/stdc++.h>
using namespace std; 
bool judge(int n)
{
    int i;
    if(n==2)
     return true;
    if(n==0||n==1)
     return false;
    else
    {
        for(i=2;i*i<=n;i++)
         if(n%i==0)
          return false;
    }
    return true;
}
int main(int argc, char** argv) 
{
    //行数n,循环变量i,j,非空格字符个数sum,列数k 
    int n,i,j,sum,k;
    cin>>n;
    for(i=0;i<n;i++)
     {
         sum=(i+1)*(i+1);//求出每一行的非空格字符总数 
         k=1;//每次循环将列数置为1 
         j=i%26;//每一次循环要输出的第一个字符 
         while(sum>=1)
          {
               if(judge(char(j+'A')+k))//判断是否为质数 
               {
                   cout<<char(j+'A');
                   j=(j+1)%26;//只有为质数才将字符向后移动 
                   sum--;//非空格字符数减一 
             }
            else
            {
                cout<<" ";//打印空格 
            }
            k++;//列数加1 
          }
        cout<<endl;
     }
    return 0;
}