问题F:画

题目描述
数学考试的时候,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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值