POJ1321 - 棋盘问题（Java代码+注释）

题目描述

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

Input

输入含有多组测试数据。
每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

Output

对于每一组数据，给出一行输出，输出摆放的方案数目C （数据保证C<2^31）。

Sample

Input

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Output

2
1

dfs

ps:前后代码为输入输出流，主要代码在TaskA中

import java.io.*;
import java.util.*;
import java.math.*;

public class Main {
    public static void main(String[] args) {
        InputStream inputStream = System.in;
        OutputStream outputStream = System.out;
        InputReader in = new InputReader(inputStream);
        PrintWriter out = new PrintWriter(outputStream);
        TaskA solver = new TaskA();
        solver.solve(1, in, out);
        out.close();
    }

    static class TaskA {
        int sum;
        int n,k;
        char[][] a = new char[10][10];
        boolean[] aa = new boolean[10];
        void dfs(int m,int ans){//ans为已经放置的棋子数量
            if(ans == k){//当所有棋子放置后方案加一
                sum++;
                return;
            }
            for (int i = m; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    if (a[i][j] == 0 && !aa[j]) {//判断该位置可放置，进行dfs并标记
                        aa[j] = true;
                        dfs(i + 1, ans + 1);//搜索下一层并棋子数减一
                        aa[j] = false;
                    }
                }
            }
        }
        public void solve(int testNumber, InputReader in, PrintWriter out) {
            while (in.hasNext()) {
                n = in.nextInt();
                k = in.nextInt();
                sum = 0;
                if (n == -1 && k == -1) {
                    break;
                }else{
                    for (int i = 0; i < n; i++) {
                        String s = in.next();
                        for (int j = 0; j < n; j++) {
                            if(s.charAt(j) == '#')
                                a[i][j] = 0;
                            else
                                a[i][j] = 1;
                        }
                    }
                    dfs(0,0);
                }
                out.println(sum);
            }
        }
    }

    static class InputReader {
        public BufferedReader reader;
        public StringTokenizer tokenizer;

        public InputReader(InputStream stream) {
            reader = new BufferedReader(new InputStreamReader(stream), 32768);
            tokenizer = null;
        }
        boolean hasNext()
        {
            while (tokenizer == null || !tokenizer.hasMoreTokens())
            {
                try
                {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (Exception e)
                {
                    return false;
                    // TODO: handle exception
                }
            }
            return true;
        }
        public String next() {
            while (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }
        public String nextLine()
        {
            String str = null;
            try
            {
                str = reader.readLine();
            } catch (IOException e)
            {
                e.printStackTrace();
            }
            return str;
        }
        public int nextInt() {
            return Integer.parseInt(next());
        }
        public double nextDouble(){
            return Double.parseDouble(next());
        }
        public long nextLong(){
            return Long.parseLong(next());
        }
        public BigInteger nextBigInteger()
        {
            return new BigInteger(next());
        }
        public BigDecimal nextBigDecimal()
        {
            return new BigDecimal(next());
        }

    }
}

题目链接

https://vjudge.net/problem/POJ-1321/origin