LeetCode: N-Queens II

这道题其实我原来的思想是对的，不过不知道vector的速度要比数组慢很多，还有就是把参数放到private里也会减少运行时间。memset的时间和一个for循环是一样的，所以以后如果能确定一个数组的size并且不会改变数组大小的，就可以不用vector而用数组。多数次过

class Solution {
private:
    int ret;
    bool visit[100];
    int rec[100];
public:
    bool check(int i, int dep) {
        for (int j = 0; j < dep; j++) {
            if (abs(j - dep) == abs(i - rec[j])) return false;
        }
        return true;
    }
    void dfs(int dep, int n) {
         if (dep == n) {
             ret++;
             return;
         }
         for (int i = 0; i < n; i++) {
             if (!visit[i] && check(i, dep)) {
                 visit[i] = true;
                 rec[dep] = i;
                 dfs(dep+1, n);
                 visit[i] = false;
             }
         }
     }
    int totalNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        ret = 0;
        memset(visit, false, sizeof(visit));
        dfs(0, n);
        return ret;
    }
};

 C#

public class Solution {
    public int TotalNQueens(int n) {
        int ans = 0;
        bool[] visit = new bool[100];
        int[] rec = new int[100];
        dfs(0, n, ref ans, ref visit, ref rec);
        return ans;
    }
    public void dfs(int dep, int n, ref int ans, ref bool[] visit, ref int[] rec) {
        if (dep == n) {
            ans++;
            return;
        }
        for (int i = 0; i < n; i++) {
            if (!visit[i] && check(i, dep, ref rec)) {
                visit[i] = true;
                rec[dep] = i;
                dfs(dep+1, n, ref ans, ref visit, ref rec);
                visit[i] = false;
            }
        }
    }
    public bool check(int i, int dep, ref int[] rec) {
        for (int j = 0; j < dep; j++) {
            if (Math.Abs(j - dep) == Math.Abs(i - rec[j])) return false;
        }
        return true;
    }
}