【leetcode】N-Queens II

class Solution {
public:
    bool check(int row, int* colArray) {
        for (int i = 0; i < row; ++i) 
        {
            int diff = abs(colArray[i] - colArray[row]);      // in a col
            if (diff == 0 || diff == row - i) {         // int a row or line
                return false;
            }
        }
        return true;
    }

    void placeQueens(int row, int n, int &count, int* colArray) {
        if (row == n) {
            ++count;
            return;
        }
        
        for (int col = 0; col < n; ++col) {     // in 0 row, test n col
            colArray[row] = col;
            if (check(row, colArray)){
                placeQueens(row+1, n, count, colArray);    // test other rows
            }
        }
    }
    
    int totalNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int *colArray = new int[n];
        int count = 0;
        
        placeQueens(0, n, count, colArray);
        
        return count;
    }
};

class Solution {
public:
    bool check(int col, int *rowArray){
        for(int i = 0; i < col; i++){
            if(rowArray[i] == rowArray[col] || abs(rowArray[i] - rowArray[col]) == col - i)
                return false;
        }
        return true;
    }
    
    void placeQueens(int col, int n, int &count, int *rowArray){
        if(col == n){
            count++;
            return;
        }
        
        for(int row = 0; row < n; row++){
            rowArray[col] = row;
            if(check(col, rowArray)){
                placeQueens(col+1, n, count, rowArray);
            }
        }
    }

    int totalNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int *rowArray = new int[n];
        int count = 0;
        
        placeQueens(0, n, count, rowArray);
        
        return count;
    }
};

    bool check(int row, int* colArray) {
        for (int i = 0; i < row; ++i) 
        {
            int diff = abs(colArray[i] - colArray[row]);        // in a col
            if (diff == 0 || diff == row - i) {                 // int a row or line
                return false;
            }
        }
        return true;
    }
    
    int totalNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        
        if(n == 1) return 1;
        
        int *colArray = new int[n+1];
        int count = 0;
        
        colArray[1] = 0;
        int k = 1;
        while(k > 0){
            
            colArray[k] += 1;
            while( (colArray[k] <= n) && !check(k, colArray) ){
                colArray[k]++;
            }
            
            if(colArray[k] <= n){
                if(k == n){
                    count++;
                } else {
                    colArray[++k] = 0;
                } 
            } else {
                k--;
            }
        }
        
        return count;
    }

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

bool check(int row, int* colArray) {
    for (int i = 0; i < row; ++i) 
    {
        int diff = abs(colArray[i] - colArray[row]);        
        if (diff == 0 || diff == row - i) {                 
            return false;
        }
    }
    return true;
}

int totalNQueens(int n) {


    if(n == 1) return 1;

    int *colArray = new int[n+1];
    int count = 0;

    colArray[1] = 0;
    int k = 1;
    while(k > 0){

        colArray[k] += 1;
        while( (colArray[k] <= n) && !check(k, colArray) ){
            colArray[k]++;
        }

        if(colArray[k] <= n){
            if(k == n){
                count++;
            } else {
                colArray[++k] = 0;
            } 
        } else {
            k--;
        }
    }


    printf("%d\t%d\n", n, count);
    return count;
}

int main(){
    int num = 16;
    for(int i = 0; i < num; i++){
        totalNQueens(i);
    }

    return 0;
}