[入試数学]阪大2017理系大問2をC言語で解いてみた

#include <stdio.h>
#include <math.h>

typedef struct complex {
    double r;
    double i;
}complex;


void cre_z(int N,complex z[]){
    for(int i=0; i<N; i++){
        z[i].r = cos(2*i*M_PI/N);
        z[i].i = sin(2*i*M_PI/N);
     
    }
}

void clear_comp(complex *w){
    w->i = 0;
    w->r = 0;
}

complex func(int a,int b,int c, int d, int e){
    complex w;
    clear_comp(&w);
    
    complex z[5];
    for(int j=0; j<5; j++)
        clear_comp(&z[j]);
    
    cre_z(5,z);
    
    w.r = a + b*z[1].r + c*z[2].r + d*z[3].r + e*z[4].r;
    w.i =b*z[1].i + c*z[2].i + d*z[3].i + e*z[4].i;
    
    return w;
}

double length(complex w){
    return sqrt((w.r*w.r)+(w.i*w.i));
}

int all_c = 0;
int ok = 0;

int main(void){
    complex w;
    
    for(int a=0; a<=1; a++){
        for(int b=0; b<=1; b++){
            for(int c=0; c<=1; c++){
                for(int d=0; d<=1; d++){
                    for(int e=0; e<=1; e++){
                        clear_comp(&w);
                        w = func(a, b, c, d, e);
                        if(length(w) < 0.99999)                        
                            ok++;
                        
                        all_c++;
                    }
                }
            }
        }
    }
    
    printf("%d/%d\n",ok,all_c);
    printf("%f\n",(double)ok/all_c);
    
    return 0;
}