码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj1087 [SCOI2005]互不侵犯King

时间:2016-12-17 11:17:54      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:color   ini   problem   ems   continue   algorithm   信息   code   last   

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 3306  Solved: 1915
[Submit][Status][Discuss]

Description

  在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。

Input

  只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)

Output

  方案数。

Sample Input

3 2

Sample Output

16
分析:这道题求方案数,那么可以想到用dp或者搜索,甚至还可以用数学方法,然而对于本题而言,搜索需要记录的信息和状态太多,会TLE,也没有什么很好的数学方法来解决这道题,所以只有采用dp.
       注意到n非常小,而且状态需要记录的信息非常多,所以考虑状态压缩dp.对于这类棋盘问题,当前行的状态和上一行的状态有着密切联系,而且答案和k有关,所以我们设计状态dp[i][j][k]为前i行已经放了j个国王并且第i行的状态为k(二进制)的方案数,那么dp[i][j][k] = Σdp[i-1][j - num[k]][p],其中num数组记录着一行为状态k的放的国王的数目,p为上一行符合要求的状态.
       现在的问题就是如何判断状态i是否符合要求,这不仅与i本身有关,还和它上一行的状态j有关,利用 <<,>>,&就可以判断了,它的原理是什么呢?我们把每个状态抽象成1个二进制数,它的第i位表示第i列放不放国王,利用&的性质,可以判断上下两行是否冲突,那么如何判断当前行是否冲突呢?将i左移1位再与i进行&操作即可.
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

int n, k,num[1000];
long long dp[10][1000][1000];
bool flag[1000];
long long ans;

void init()
{
    for (int i = 0; i < (1 << n); i++)
        if (!(i & (i << 1)))
        {
            flag[i] = true;
            int t = i;
            while (t)
            {
                num[i] += (t & 1);
                t >>= 1;
            }
            dp[1][num[i]][i] = 1;
        }
}

int main()
{
    scanf("%d%d", &n, &k);
    init();
    for (int i = 2; i <= n; i++)
        for (int j = 0; j <= k; j++)
            for (int now = 0; now < (1 << n); now++)
            {
                if (!flag[now])
                    continue;
                if (num[now] > j)
                    continue;
                for (int last = 0; last < (1 << n); last++)
                {
                    if (!flag[last])
                        continue;
                    if ((last & now) || ((now << 1) & last) || ((now >> 1) & last))
                        continue;
                    dp[i][j][now] += dp[i - 1][j - num[now]][last];
                }
            }
    for (int i = 0; i < (1 << n); i++)
        ans += dp[n][k][i];

    printf("%lld", ans);
    return 0;
}

 

bzoj1087 [SCOI2005]互不侵犯King

标签:color   ini   problem   ems   continue   algorithm   信息   code   last   

原文地址:http://www.cnblogs.com/zbtrs/p/6189240.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!