开灯问题 C++实现

    int a=0;
    int b;
    cin >> a;
    cin >> b;
    int arr[max];
    for (int j = 0; j < a; j++) {
        arr[j] = 1;
    }
    for (int i = 2; i <= b; i++) {
        for (int z = 0; z < a; z++) {
            if ((z + 1) % i == 0) {
                if (arr[z] == 1) {
                    arr[z] = 0;
                }
                else {
                    arr[z] = 1;
                }
            }
        }
        }
    for (int i = 0; i < a; i++) {
        if (arr[i] == 1)
            cout << i + 1 << " ";
    }