题目大意
定义一个长度为
n
n
n的数集序列
S
:
{
s
1
,
⋯
,
s
n
}
S:\{s_1,\cdots,s_n\}
S:{s1,⋯,sn}是好的,当且仅当
∀
i
∈
[
1
,
n
−
1
]
,
∣
s
i
/
s
i
−
1
∣
=
1
或
∣
s
i
−
1
/
s
i
∣
=
1
\forall i\in[1,n-1],|s_i/s_{i-1}|=1或|s_{i-1}/s_{i}|=1
∀i∈[1,n−1],∣si/si−1∣=1或∣si−1/si∣=1,
s
i
⊂
[
1
,
m
]
s_i\subset[1,m]
si⊂[1,m],定义一个数集序列
S
S
S的贡献为
∏
i
=
1
m
∑
j
=
1
n
[
i
∈
s
j
]
\prod_{i=1}^m\sum_{j=1}^n[i\in s_j]
∏i=1m∑j=1n[i∈sj],求所有数集序列的贡献的和,对
998244353
998244353
998244353取模
n
,
m
≤
200000
n,m\le200000
n,m≤200000
题解
设
x
i
x_i
xi表示只在
s
i
s_i
si和
s
i
+
1
s_{i+1}
si+1中出现一次的元素
现在考虑已知
x
i
,
i
∈
[
1
,
n
−
1
]
x_i,i\in[1,n-1]
xi,i∈[1,n−1],怎么求答案,需要考虑每个元素一开始是否在
s
1
s_1
s1中,发现各个元素之间互不影响,且若
a
i
a_i
ai为
s
1
s_1
s1中有
i
i
i,
i
i
i在所有集合中出现的个数,则
s
1
s_1
s1中没
i
i
i,
i
i
i在所有集合中出现的个数为
n
−
a
i
n-a_i
n−ai,所以有贡献和为
∏
i
=
1
m
(
a
i
+
n
−
a
i
)
=
n
m
\prod_{i=1}^m(a_i+n-a_i)=n^m
∏i=1m(ai+n−ai)=nm,可以发现
x
x
x是什么与答案并无关系,故再乘上
x
x
x的方案数
m
n
−
1
m^{n-1}
mn−1即为答案
code \text{code} code
#include<cstdio>
#define ll long long
using namespace std;
const ll mod=998244353;
ll ksm(ll a,ll b)
{
if(b==0) return 1;
ll tmp=ksm(a,b>>1);
if(b&1) return tmp*tmp%mod*a%mod;
else return tmp*tmp%mod;
}
int n,m;
int main()
{
scanf("%d %d",&n,&m);
printf("%lld\n",ksm(n,m)*ksm(m,n-1)%mod);
return 0;
}