正方形_找规律

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

double z[10000];
int p[10000];

int main()
{
    int a,b;
    int cou=0;
    double suma=0;
    double sumb=0;
    double sumc=0;

    scanf("%d%d",&a,&b);
    double t=a;
    for(int i=0;i<b;i++){
        for(int j=0;j<4;j++){
            z[cou++]=t;
        }
        t=sqrt((t/2)*(t/2)*2);
    }
    cou=0;
    for(int i=0;i<1001;i++){
        for(int j=1;j<=3;j++){
            p[cou++]=j;
        }
    }
    for(int i=0;i<b*4;i++){
        if(p[i]==1){
            suma+=z[i];
            continue;
        }
        if(p[i]==2){
            sumb+=z[i];
            continue;
        }
        if(p[i]==3){
            sumc+=z[i];
            continue;
        }
    }
    printf("%.3f %.3f %.3f",suma,sumb,sumc);
    return 0;
}