[ZJOI2008]生日聚会 题解

[ZJOI2008]生日聚会 题解

题目地址:洛谷:【P2592】[ZJOI2008]生日聚会 – 洛谷、BZOJ:Problem 1037. — [ZJOI2008]生日聚会Party

题目描述

今天是hidadz小朋友的生日,她邀请了许多朋友来参加她的生日party。 hidadz带着朋友们来到花园中,打算坐成一排玩游戏。为了游戏不至于无聊,就座的方案应满足如下条件:
对于任意连续的一段,男孩与女孩的数目之差不超过k。
很快,小朋友便找到了一种方案坐了下来开始游戏。hidadz的好朋友Susie发现,这样的就座方案其实是很多的,所以大家很快就找到了一种,那么到底有多少种呢?热爱数学的hidadz和她的朋友们开始思考这个问题……
假设参加party的人中共有n个男孩与m个女孩,你是否能解答Susie和hidadz的疑问呢?由于这个数目可能很多,他们只想知道这个数目除以12345678的余数。

题意简述

有$n$个男孩$m$个女孩,要求把这些排成一排,使得每一个连续区间内男女孩的数量之差的绝对值不超过$k$,求排列的方案数,对$12345678$取模。

输入输出格式

输入格式:
输入文件party.in仅包含一行共3个整数,分别为男孩数目n, 女孩数目m, 常数k。

输出格式:
输出文件party.out应包含一行,为题中要求的答案。

输入输出样例

输入样例#1:

1 2 1

输出样例#1:

1

说明

对于30%的数据,n , m ≤ 20;
对于100%的数据, n , m ≤ 150,k ≤ 20。

题解

大力DP。
设计状态$dp[i][j][x][y]$表示排列前$i$个人,其中$j$个男孩,$i-j$个女孩,以$i$结尾的任意一段男孩-女孩的最大值$x$,以$i$结尾的任意一段女孩-男孩的最大值$y$的方案数,若$x, y$出现负数情况视为$0$处理,大力枚举第$i$个是男孩还是女孩转移即可。
复杂度$O(n^2k^2)$。

代码

// Code by KSkun, 2018/7
#include <cstdio>
#include <cctype>
#include <cstring>

#include <algorithm>

typedef long long LL;

inline char fgc() {
    static char buf[100000], *p1 = buf, *p2 = buf;
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2)
        ? EOF : *p1++;
}

inline LL readint() {
    register LL res = 0, neg = 1; register char c = fgc();
    for(; !isdigit(c); c = fgc()) if(c == '-') neg = -1;
    for(; isdigit(c); c = fgc()) res = (res << 1) + (res << 3) + c - '0';
    return res * neg;
}

const int MO = 12345678;

int n, m, k, dp[305][155][25][25];

int main() {
    n = readint(); m = readint(); k = readint();
    dp[0][0][0][0] = 1;
    for(int i = 0; i < n + m; i++) {
        for(int j = 0; j <= n; j++) {
            for(int x = 0; x <= k; x++) {
                for(int y = 0; y <= k; y++) {
                    if(!dp[i][j][x][y]) continue;
                    if(j + 1 <= n && x + 1 <= k) {
                        dp[i + 1][j + 1][x + 1][std::max(y - 1, 0)] += dp[i][j][x][y];
                        dp[i + 1][j + 1][x + 1][std::max(y - 1, 0)] %= MO;
                    }
                    if(i + 1 - j <= m && y + 1 <= k) {
                        dp[i + 1][j][std::max(x - 1, 0)][y + 1] += dp[i][j][x][y];
                        dp[i + 1][j][std::max(x - 1, 0)][y + 1] %= MO;
                    }
                }
            }
        }
    }
    int ans = 0;
    for(int i = 0; i <= k; i++) {
        for(int j = 0; j <= k; j++) {
            ans = (ans + dp[n + m][n][i][j]) % MO;
        }
    }
    printf("%d", ans);
    return 0;
}


发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据