JZOJ-senior-5923. 【NOIP2018模拟10.23】Bomb

xiaoxiao2025-06-02  47

Time Limits: 1000 ms Memory Limits: 524288 KB

Description

常数国与 Hack 国近年来战火纷飞。

常数国共有 n 个城市,每两个城市之间均有一条交通线联通。如今常数国遭到 Hack 国的重创,岌岌可危。Hack 国国王小 K 决定轰炸常数国的交通线,对常数国发起最后的攻击。

面对危难之时,常数国国王决定更换首都。在 Hack 国的轰炸结束之后,常数国的领土将会分成若干个联通块。常数国的首都,将会从联通块大小最大的联通块中,随机选择一个城市,作为首都。

小 K 得知了常数国的应对方案之后,他想知道,Hack 国有多少种不同的轰炸方案,使得常数首都所在的联通块大小恰好为 k。两种轰炸方案是不同的,当且仅当一条交通线在一种方案中存在,在另一种方案中被轰炸。由于方案数可能很大,你需要输出方案数对 998,244,353 取模的结果。

Input

从文件bomb.in中读入数据。 共一行,两个整数 n,k ,表示常数国城市的个数与首都所在联通块大小。

Output

输出到文件bomb.out中。 共一行,表示 Hack 国的轰炸方案数对 998,244,353 取模后的结果。

Sample Input

​Sample Input 1 3 2

Sample Input 2 5 3

Sample 3 见选手目录下的bomb/bomb3.in与bomb/bomb3.ans。 该组样例的数据范围同第 8 个测试点。

Sample Output

Sample Output1 3

Explanation 3 种方案分别为,仅保留 1 号城市与 2 号城市的交通线,仅保留 2 号城市与3 号城市的交通线,仅保留 1 号城市与 3 号城市的交通线。

Sample Output2 80

Data Constraint

对于 100% 的数据,满足 1 ≤ k ≤ n ≤ 2 × 10 3 。除此之外,对于每个数据点,还满足以下限制。

Solution

套路题,然而我还是不会 设 f [ i ] f[i] f[i] 表示大小为 i i i 的保证联通的图的个数 那么怎么求 f [ i ] f[i] f[i] 呢?正难则反!!!用总数减去不连通的情况数

h [ i ] h[i] h[i] 表示大小为 i i i 的不保证联通的图的个数,显然有 h [ i ] = 2 C i 2 h[i]=2^{C^2_i} h[i]=2Ci2 为了不重不漏,我们将枚举的 j j j 强制表示为编号最小的节点所在联通块的大小 那么 f [ i ] f[i] f[i] 可以写成这样 f [ i ] = h [ i ] − ∑ j = 1 i − 1 C i − 1 j − 1 ∗ f [ i ] ∗ h [ i − j ] f[i]=h[i]-\sum_{j=1}^{i-1}C_{i-1}^{j-1}*f[i]*h[i-j] f[i]=h[i]j=1i1Ci1j1f[i]h[ij]

g [ i ] g[i] g[i] 表示大小为 i i i 的图中最大联通块大小不超过 i i i 的图的个数 于是有如下转移 g [ i ] = ∑ j = 1 m i n ( i , k ) C i − 1 j − 1 ∗ f [ j ] ∗ g [ i − j ] g[i]=\sum_{j=1}^{min(i,k)}C_{i-1}^{j-1}*f[j]*g[i-j] g[i]=j=1min(i,k)Ci1j1f[j]g[ij]

由于 g g g 表示最大联通块大小 &lt; = k &lt;=k <=k 的情况 我们类似地设一个 g 1 g1 g1 表示最大联通块大小 &lt; = k − 1 &lt;=k-1 <=k1 的情况 转移方程就是将上方的 k k k 换成 k − 1 k-1 k1 就好了

最终答案就是 g [ n ] − g 1 [ n ] g[n]-g1[n] g[n]g1[n] 时间复杂度 O ( N 2 ) O(N^2) O(N2)

Code

#include<algorithm> #include<cstdio> #define fo(i,a,b) for(int i=a;i<=b;++i) #define fd(i,a,b) for(int i=a;i>=b;--i) #define ll long long using namespace std; const int N=2e3+5,P=998244353; int n,k; ll h[N],f[N],g[N],g1[N],_2[N*N>>1],C[N][N]; int main() { freopen("bomb.in","r",stdin); freopen("bomb.out","w",stdout); scanf("%d%d",&n,&k); fo(i,0,n) C[i][0]=1; fo(i,1,n) fo(j,1,i) C[i][j]=(C[i-1][j-1]+C[i-1][j])%P; _2[0]=1; fo(i,1,n*(n-1)>>1) _2[i]=((ll)_2[i-1]<<1)%P; h[0]=f[0]=g[0]=g1[0]=1; fo(i,1,n) h[i]=_2[i*(i-1)>>1]; fo(i,1,n) { fo(j,1,i-1) f[i]=(f[i]+C[i-1][j-1]*f[j]%P*h[i-j]%P)%P; f[i]=(h[i]-f[i]+P)%P; fo(j,1,min(i,k)) g[i]=(g[i]+C[i-1][j-1]*f[j]%P*g[i-j]%P)%P; fo(j,1,min(i,k-1)) g1[i]=(g1[i]+C[i-1][j-1]*f[j]%P*g1[i-j]%P)%P; } printf("%lld",(g[n]-g1[n]+P)%P); }
转载请注明原文地址: https://www.6miu.com/read-5031152.html

最新回复(0)