#include <stdio.h>
#include <stdlib.h>
#include <time.h>

void main() {
        int a, b, ans, temp, op;
        srand(time(NULL));
        while(1) {

                a = rand()%9 +1;
                b = rand()%9 +1;
                op = rand()%4 +1;

                if(op == 1) {
                        printf("%d + %d = ", a, b);
                        scanf("%d", &ans);
                        if (a+b == ans) { printf("정답\n"); }
                        if (a+b != ans) { printf("오답\n"); }
                }

                if(op == 2) {
                        if(a > b) {
                                temp = a;
                                a = b;
                                b = temp;
                        }
                        printf("%d - %d = ", a+b, b);
                        scanf("%d", &ans);
                        if ((a+b)-b == ans) { printf("정답\n"); }
                        if ((a+b)-b != ans) { printf("오답\n"); }
                }

                if(op == 3) {
                        printf("%d * %d = ", a, b);
                        scanf("%d", &ans);
                        if (a*b == ans) { printf("정답\n"); }
                        if (a*b != ans) { printf("오답\n"); }
                }

                if(op == 4) {
                        printf("%d / %d = ", (a*b), b);
                        scanf("%d", &ans);
                        if ((a*b)/b == ans) { printf("정답\n"); }
                        if ((a*b)/b != ans) { printf("오답\n"); }
                }
        }
}



이걸 switch case만 써서 바꿀수 있나요?