LightOJ 1005 Rooks 动态规划dp || 组合数学

最新推荐文章于 2020-12-27 22:58:44 发布

原创最新推荐文章于 2020-12-27 22:58:44 发布 · 654 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#动态规划 #lightoj #组合数学

乱搞dp 同时被 2 个专栏收录

17 篇文章

订阅专栏

组合数学

2 篇文章

订阅专栏

博客详细介绍了如何解决LightOJ的1005题——Rooks问题。在n*n的棋盘上放置k个棋子，使得任意两个棋子不互相攻击的方案数。提供了两种解法，一种是动态规划（dp），状态转移方程为dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * (n- (j-1))；另一种是组合数学的方法，利用组合公式并考虑重复枚举的情况。

题目：http://www.lightoj.com/volume_showproblem.php?problem=1005

题意：给定一个n * n的棋盘，往上面放k个棋子，棋子可以攻击所在的行或列，求把k个棋子放在棋盘上使任意两个不能互相攻击的方案数

思路：有两种解法，dp或者组合数学

一：dp方法

设dp[i][j]为前i行放j个棋子时满足条件的方案数，dp[i][0]置为1，可以得到状态转移方程dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * (n- (j-1))。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define debug() puts("here")
using namespace std;

typedef long long ll;
const int N = 110;
int cas;
ll dp[N][N];
int main()
{
    int t, n, k;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &k);
        memset(dp, 0, sizeof dp);
        dp[0][0] = 1;
        for(int i = 1; i <= n; i++)
            for(int j = 0; j <= k; j++)
                if(j) dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * (n - (j-1));
                else dp[i][j] = dp[i-1][j];
        printf("Case %d: %lld\n", ++cas, dp[n][k]);
    }
    return 0;
}

二：组合数学

首先说我自己的思路：放一个棋子，无论放在哪里，那么方案数是C(n, 1)，然后有一行一列不能放了，意味着棋盘可用空间为（n-1） * (n-1)，然后就是C(n-1, 1), C(n-2, 1)...直到放完为止。又防止的时候用的是组合，会出现重复枚举，所以再除以k!

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define debug() puts("here")
using namespace std;

typedef long long ll;
const int N = 110;
int cas;
bool vis[1010];
void slove(int n, int k)
{
    ll sum = 1;
    memset(vis, 0, sizeof vis);
    for(int i = n - k + 1; i <= n; i++)
    {
        sum *= i * i;
        for(int j = 2; j <= k; j++)
            if(sum % j == 0 && !vis[j])
                sum /= j, vis[j] = true;
    }
    printf("Case %d: %lld\n", ++cas, sum);
}
int main()
{
    int t, n, k;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &k);
        if(k > n) printf("Case %d: %d\n", ++cas, 0);
        else slove(n, k);
    }
    return 0;
}

然后这是我看的别人的。首先从n行中选出k行，就是C(n, k)，然后从n列中选出k列，这个是排列，是A(n, k)，所以答案就是C(n, k) * A(n, k)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define debug() puts("here")
using namespace std;
 
typedef long long ll;
const int N = 110;
int cas;
bool vis[1010];
void slove(int n, int k)
{
    ll sum = 1;
    for(int i = n - k + 1; i <= n; i++)
        sum *= i;
    ll ans = sum;
    for(int i = 1; i <= k; i++)
        sum /= i;
    printf("Case %d: %lld\n", ++cas, sum * ans);
}
int main()
{
    int t, n, k;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &k);
        if(k > n) printf("Case %d: %d\n", ++cas, 0);
        else slove(n, k);
    }
    return 0;
}