[HDU4532]湫秋系列故事——安排座位

题面在这里

description

有\(n\)种颜色的小球,每种颜色的小球有\(a_i\)个;

要把它们摆成一排,求相邻小球颜色不相同的摆放方案数。

任意两个合理的安排方法,只要有一个位置的同学不同,都被认为是不同的。

data range

\[n\le 50,a_i\le 50,\sum a_i\le 500\]

solution

鸣谢YCB

以下先考虑可重排列的情况

考虑容斥:

\[Ans=\sum_{i=0}^{n-1}(-1)^{i}f(i)\]

其中\(f(x)\)表示至少有\(x\)对相邻同色球的方案数。

直接求多项式\(f(x)\)不好求,我们先考虑一种颜色。

由于可重排列的方案数\(P=\frac{(\sum a_i)!}{\sum(a_i!)}\),

对于一种颜色的小球,构造多项式

\[g_i(x)=\sum_{j=0}^{a_i-1}(-1)^{j}\frac{1}{(a_i-j)!}E(a_i,a_i-j)x^{j}\]

其中\(E(i,j)\)表示把\(i\)个相同的小球分成\(j\)段的方案数,

相当于在\(i-1\)个空隙中选\(j-1\)块隔板,因此\(E(i,j)=\binom{i-1}{j-1}\)。

于是\[f(x)=(\sum_{i=1}^{n} a_i)\prod_{i=1}^{n}g_i(x)\]

直接暴力多项式乘法即可,复杂度为\(O(n(∑a_i)a_{max})\)

code

#include<bits/stdc++.h>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<iomanip>
#include<cstring>
#include<complex>
#include<vector>
#include<cstdio>
#include<string>
#include<bitset>
#include<ctime>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define FILE "a"
#define mp make_pair
#define pb push_back
#define RG register
#define il inline
using namespace std;
typedef unsigned long long ull;
typedef vector<int>VI;
typedef long long ll;
typedef double dd;
const dd eps=1e-10;
const int mod=1e9+7;
const int N=505;
const dd pi=acos(-1);
il ll read(){
    RG ll data=0,w=1;RG char ch=getchar();
    while(ch!=‘-‘&&(ch<‘0‘||ch>‘9‘))ch=getchar();
    if(ch==‘-‘)w=-1,ch=getchar();
    while(ch<=‘9‘&&ch>=‘0‘)data=data*10+ch-48,ch=getchar();
    return data*w;
}

il void file(){
    freopen(FILE".in","r",stdin);
    freopen(FILE".out","w",stdout);
}

il void upd(int &a,int b){a+=b;if(a>=mod)a-=mod;}
il void dec(int &a,int b){if(!b)return;upd(a,mod-b);}
il int poww(int a,int b){
    RG int ret=1;
    for(;b;b>>=1,a=1ll*a*a%mod)
        if(b&1)ret=1ll*ret*a%mod;
    return ret;
}
int fac[N],inv[N],C[N][N];
il void init(){
    fac[0]=1;for(RG int i=1;i<=500;i++)fac[i]=1ll*fac[i-1]*i%mod;
    inv[500]=poww(fac[500],mod-2);
    for(RG int i=499;i;i--)inv[i]=1ll*inv[i+1]*(i+1)%mod;
    C[0][0]=1;
    for(RG int i=1;i<=500;i++)
        for(RG int j=0;j<=i;j++){
            C[i][j]=C[i-1][j];
            if(j)upd(C[i][j],C[i-1][j-1]);
        }
}

int T,n,x,f[N],g[N],h[N],sum,ans,tot;
int main()
{
    init();T=read();
    for(RG int t=1;t<=T;t++){
        n=read();ans=sum=0;memset(f,0,sizeof(f));f[0]=tot=1;
        for(RG int i=1;i<=n;i++){
            x=read();sum+=x;tot=1ll*tot*fac[x]%mod;
            for(RG int j=0;j<x;j++){
                g[j]=1ll*inv[x-j]*C[x-1][x-j-1]%mod;
                if((j&1)&&g[j])g[j]=mod-g[j];
            }
            memset(h,0,sizeof(h));
            for(RG int j=0;j<sum-x-i+2;j++)
                for(RG int k=0;k<x;k++)
                    upd(h[j+k],1ll*f[j]*g[k]%mod);
            swap(f,h);
        }
        for(RG int i=0;i<sum-n+2;i++)
            upd(ans,1ll*f[i]*fac[sum-i]%mod);
        printf("Case %d: %lld\n",t,1ll*ans*tot%mod);
    }
    return 0;
}

原文地址:https://www.cnblogs.com/cjfdf/p/9151547.html

时间: 2024-10-10 10:21:49

[HDU4532]湫秋系列故事——安排座位的相关文章

HDU-4532湫秋系列故事——安排座位

Problem Description 为了给腾讯公司找到更多优秀的人才,HR湫秋最近去某高校组织了一次针对该校所有系的聚会,邀请了每个系的一些优秀学生来参加. 作为组织者,湫秋要安排他们的座位.这并不是一件很简单的事情,因为只有一排位置,并且位置总数恰好等于参加聚会的人数.为了促进交流,两个来自相同系的同学不可以座位相邻.湫秋现在希望知道有多少种不同的合理安排座位的方法(任意两个合理的安排方法,只要有一个位置的同学不同,都被认为是不同的). Input 输入第一行为T,表示有T组测试数据. 每

hdu----(4522)湫湫系列故事——过年回家(最短路)

湫湫系列故事——过年回家 Time Limit: 500/200 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)Total Submission(s): 1095    Accepted Submission(s): 226 Problem Description 出门在外,最想念的还是家,对在深圳腾讯工作的HR湫湫来说,春节回家是一年中最期盼的事,不仅可以见到阔别已久的亲人,还能以相亲的名义调侃众多帅哥(她的内心告诉她

湫湫系列故事——减肥记I(杭电4508)(完全背包)

湫湫系列故事--减肥记I Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) Total Submission(s): 2612    Accepted Submission(s): 1345 Problem Description 对于吃货来说,过年最幸福的事就是吃了,没有之一! 但是对于女生来说,卡路里(热量)是天敌啊! 资深美女湫湫深谙"胖来如山倒,胖去如抽丝"的道

HDU4508 湫湫系列故事——减肥记I

这题一开始看还以为是0,1背包问题,不过老师说过了完全背包就按照这个思路做.(虽然偷偷去百度了一下到底是不是,有点怀疑老师是不是记错了的说); 顺便记下对一维数组方程的理解; for(int i =1;i<=n;i++)有n次就可以啦;//闭区间 for(int j=b[i];j<=m;j++)//每进行一次循环,第i件物品数加1或者0; dp[j]=max(dp[j],dp[j-b[i]]+a[i]);//如果是最优解的话,先后顺序无所谓是不是; 下面给出代码.嘻嘻. //这个在背包九讲里面

HDU4508 湫湫系列故事——减肥记I 【完全背包】

湫湫系列故事--减肥记I Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) Total Submission(s): 2458    Accepted Submission(s): 1254 Problem Description 对于吃货来说,过年最幸福的事就是吃了,没有之一! 但是对于女生来说,卡路里(热量)是天敌啊! 资深美女湫湫深谙"胖来如山倒,胖去如抽丝"的道

HDU 4508 湫湫系列故事——减肥记I (2013腾讯编程马拉松初赛第一场)

http://acm.hdu.edu.cn/showproblem.php?pid=4508 题目大意: 给定一些数据. 每组数据以一个整数n开始,表示每天的食物清单有n种食物. 接下来n行,每行两个整数a和b,其中a表示这种食物可以带给湫湫的幸福值(数值越大,越幸福),b表示湫湫吃这种食物会吸收的卡路里量. 最后是一个整数m,表示湫湫一天吸收的卡路里不能超过m. 思路: 完全背包. 一开始以为是01背包. 敲了01后样例2不对啊!!! 然后改成完全就过了..就改循环体就好了.. #includ

HDU 4514 湫湫系列故事——设计风景线(并查集+树形DP)

湫湫系列故事——设计风景线 Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) Total Submission(s): 4669    Accepted Submission(s): 853 Problem Description 随着杭州西湖的知名度的进一步提升,园林规划专家湫湫希望设计出一条新的经典观光线路,根据老板马小腾的指示,新的风景线最好能建成环形,如果没有条件建成环形

G - 湫湫系列故事——减肥记I

G - 湫湫系列故事——减肥记I Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u Description 对于吃货来说,过年最幸福的事就是吃了,没有之一!  但是对于女生来说,卡路里(热量)是天敌啊!  资深美女湫湫深谙“胖来如山倒,胖去如抽丝”的道理,所以她希望你能帮忙制定一个食谱,能使她吃得开心的同时,不会制造太多的天敌. 当然,为了方便你制作食谱,湫湫给了你每日食物清单,上面描述了当天

hdu-----(4514)湫湫系列故事——设计风景线(树形DP+并查集)

湫湫系列故事——设计风景线 Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)Total Submission(s): 3105    Accepted Submission(s): 562 Problem Description 随着杭州西湖的知名度的进一步提升,园林规划专家湫湫希望设计出一条新的经典观光线路,根据老板马小腾的指示,新的风景线最好能建成环形,如果没有条件建成环形,