实现朴素贝叶斯分类器

该题一个麻烦的是输入与输出的格式的转换 由于不能用第三方库 只能转成list形式 

# 看到题的时候没想到什么好的解决办法，只能算出每个概率，然后计算后验概率，如果有大佬有好的解决办法就教教孩子吧
#include<iostream>
#include<vector>
#include<math.h>
using namespace std;
int main(){
    float A[3] = {0}, B[3] = {0}, C[2] = {0}, D[2] = {0}, Y[2] = {0}, AY[3][2] = {0}, BY[3][2] = {0}, CY[2][2] = {0}, DY[2][2] = {0};
    int M; cin >> M;
    int train[M][5];
    for(int i=0; i<M; i++){
        for(int j=0; j<5; j++){
            cin >> train[i][j];
            if(j==0){
                A[train[i][j]-1]++;
            }
            if(j==1){
                B[train[i][j]-1]++;
            }
            if(j==2){
                C[train[i][j]]++;
            }
            if(j==3){
                D[train[i][j]]++;
            }
            if(j==4){
                Y[train[i][j]]++;
                AY[train[i][0]-1][train[i][j]]++;
                BY[train[i][1]-1][train[i][j]]++;
                CY[train[i][2]][train[i][j]]++;
                DY[train[i][3]][train[i][j]]++;
            }
        }
    }
    A[0] = A[0]/double(M); A[1] = A[1]/double(M); A[2] = A[2]/double(M); 
    B[0] = B[0]/double(M); B[1] = B[1]/double(M); B[2] = B[2]/double(M); 
    C[0] = C[0]/double(M); C[1] = C[1]/double(M); 
    D[0] = D[0]/double(M); D[1] = D[1]/double(M); 
    
    //AY,BY,CY,DY
    AY[0][0] = AY[0][0]/double(Y[0]); AY[1][0] = AY[1][0]/double(Y[0]); AY[2][0] = AY[2][0]/double(Y[0]); 
    AY[0][1] = AY[0][1]/double(Y[1]); AY[1][1] = AY[1][1]/double(Y[1]); AY[2][1] = AY[2][1]/double(Y[1]);
    BY[0][0] = BY[0][0]/double(Y[0]); BY[1][0] = BY[1][0]/double(Y[0]); BY[2][0] = BY[2][0]/double(Y[0]); 
    BY[0][1] = BY[0][1]/double(Y[1]); BY[1][1] = BY[1][1]/double(Y[1]); BY[2][1] = BY[2][1]/double(Y[1]);
    CY[0][0] = CY[0][0]/double(Y[0]); CY[1][0] = CY[1][0]/double(Y[0]); 
    CY[0][1] = CY[0][1]/double(Y[1]); CY[1][1] = CY[1][1]/double(Y[1]); 
    DY[0][0] = DY[0][0]/double(Y[0]); DY[1][0] = DY[1][0]/double(Y[0]); 
    DY[0][1] = DY[0][1]/double(Y[1]); DY[1][1] = DY[1][1]/double(Y[1]); 
    Y[0] = Y[0]/double(M); Y[1] = Y[1]/double(M); 
    //cout << Y[0] <<' ' << Y[1] << endl;
    int N; cin >> N;
    int a, b, c, d;
    float Y0 = 0, Y1 = 0;
    for(int i=0;i<N;i++){
        cin >> a >> b >> c >> d;
        Y0 = Y[0]*(AY[a-1][0]*BY[b-1][0]*CY[c][0]*DY[d][0])/(A[a-1]*B[b-1]*C[c]*D[d]);
        Y1 = Y[1]*(AY[a-1][1]*BY[b-1][1]*CY[c][1]*DY[d][1])/(A[a-1]*B[b-1]*C[c]*D[d]);
        if(Y0>=Y1){
            cout << 0 << ' ';
        }else{
            cout << 1 << ' ';
        }
    }
    
    return 0;
}