题目描述
数学考试的时候,pac 莫名其妙地开始在草稿纸上画画。
Pac 画了一个 n 边形 ,顶点编号为 1−n,每条边长都不相等。她想把每个顶点都染成 1−c 中某一颜色,且相邻顶点颜色不能相同。
闲得无聊的她又定义了以下的操作 :
1 x p:表示第 x 个顶点必须染颜色 p。
2 x p:表示第 x 个顶点必须不染颜色 p。
3 x y:表示更改第 x 个顶点与第 y 个顶点之间边的属性,第 x 个顶点与第 y 个顶点颜色可以且必须相同(保证 y=x±1,故当 x=1时,y≠n,当 x=n 时,y≠1)。
pac 想知道所有可行的方案共有多少种。
由于结果可能过大,你只需输出它对 998244353 取模的结果即可。
输入
第一行,三个正整数 n,m,c,表示多边形边数、操作个数、颜色个数。
第 2−(m+1) 行,每三个正整数表示一个操作,具体意义见题目描述。
输出
一行一个整数,为所有可行操作和模上 998244353 的结果。
样例输入 Copy
5 2 5
2 3 4
3 2 3
样例输出 Copy
208
代码如下。
首先,先把所有的连接到一起的处理一下,对于每个连通块,
看作一个整体,即只对父亲节点做处理,其余直接跳过
然后,枚举第一个的每种颜色,对应处理出来最后一个对应的颜色
然后减去最后一个和你枚举的颜色相同的情况,就可得出答案
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+5;
typedef long long ll;
int fa[maxn];
ll dp[maxn][26];
const ll mod=998244353;
struct abc
{
int a,t,p;
}x[maxn];
bool cmp(abc a,abc b)
{
return a.a>b.a;
}
int findd(int a)
{
return fa[a]==a?a:fa[a]=findd(fa[a]);
}
void init(int n)
{
for(int i=1;i<=n;i++) fa[i]=i;
}
void mix(int a,int b)
{
int f1=findd(a),f2=findd(b);
if(f1>f2) swap(f1,f2);
fa[f2]=f1;
}
int main()
{
int n,m,c;
scanf("%d%d%d",&n,&m,&c);
init(n);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&x[i].a,&x[i].t,&x[i].p);
}
sort(x+1,x+1+m,cmp);
for(int i=1;i<=m;i++)
{
if(x[i].a==3)
mix(x[i].p,x[i].t);
else if(x[i].a==2)
{
int fx=findd(x[i].t);
dp[fx][x[i].p]=-1;
}
else
{
int fx=findd(x[i].t);
if(dp[fx][x[i].p]==-1)
{
puts("0");
return 0;
}
for(int j=1;j<=c;j++)
{
if(j==x[i].p) continue;
dp[fx][j]=-1;
}
}
}
ll ans=0;
for(int z=1;z<=c;z++)
{
if(dp[1][z]==-1) continue;
dp[1][0]=1;
dp[1][z]=1;
for(int i=2;i<=n;i++)
{
if(fa[i]==i)
{
dp[i][0]=0;
int la=findd(i-1);
//memset(dp[i],0,sizeof dp[i]);
for(int j=1;j<=c;j++)
{
if(dp[i][j]==-1) continue;
dp[i][j]=dp[la][0]-max(dp[la][j],(ll)0)+mod;
dp[i][j]%=mod;
dp[i][0]+=dp[i][j];
dp[i][0]%=mod;
}
}
}
int nn=fa[n];
if(nn==1) dp[nn][z]=0;
ans+=dp[nn][0]-max((ll)0,dp[nn][z])+mod;
ans%=mod;
dp[1][z]=0;
}
printf("%lld\n",ans%mod);
return 0;
}