ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

NOIP 模拟 $89\; \rm 排水系统$

2021-11-05 06:33:00  阅读:119  来源: 互联网

标签:ch NOIP int res ++ 89 rm define MOD


题解 \(by\;zj\varphi\)

先 \(dp\) 出来不删任何边时的答案,然后考虑去掉一条边的影响。

去掉一条边 \((u,v)\),会使得流到 \(v\) 的流量减少,但流到 \(u\) 的流量不变,就相当于是减少 \(v\) 的流量,增加 \(u\) 其它终点的流量。

发现 增加/减少 流量可以转换为初始的时候管道就有 \(+x/-x\) 的污水,所以先处理出来初始有的污水,再来一次拓扑排序即可。

但这样还是 \(n^2\) 的,\(v\) 增加的污水可以转换到 \(u\) 增加的污水,这样就可做了。

Code
#include<bits/stdc++.h>
#define ri signed
#define pd(i) ++i
#define bq(i) --i
#define func(x) std::function<x>
namespace IO{
    char buf[1<<21],*p1=buf,*p2=buf;
    #define gc() p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?(-1):*p1++
    #define debug1(x) std::cerr << #x"=" << x << ' '
    #define debug2(x) std::cerr << #x"=" << x << std::endl
    #define Debug(x) assert(x)
    struct nanfeng_stream{
        template<typename T>inline nanfeng_stream &operator>>(T &x) {
            bool f=false;x=0;char ch=gc();
            while(!isdigit(ch)) f|=ch=='-',ch=gc();
            while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=gc();
            return x=f?-x:x,*this;
        }
    }cin;
}
using IO::cin;
namespace nanfeng{
    #define FI FILE *IN
    #define FO FILE *OUT
    template<typename T>inline T cmax(T x,T y) {return x>y?x:y;}
    template<typename T>inline T cmin(T x,T y) {return x>y?y:x;}
    using ll=long long;
    static const int N=2.5e5+7,MOD=998244353;
    struct edge{int v,nxt;}e[N<<1];
    int first[N],deg[N],td[N],ot[N],a[N<<1],inv[N<<1],U[N<<1],V[N<<1],INV,t=1,n,m,r,k;
    ll dp[N],tdp[N],sum;
    std::queue<int> que;
    auto add=[](int u,int v) {e[t]={v,first[u]},first[u]=t++;};
    auto fpow=[](ll x,int y) {
        ll res=1;
        while(y) {
            if (y&1) res=res*x%MOD;
            x=x*x%MOD;
            y>>=1;
        }
        return res;
    };
    inline int main() {
        FI=freopen("water.in","r",stdin);
        FO=freopen("water.out","w",stdout);
        cin >> n >> m >> r >> k;
        for (ri i(1);i<=k;pd(i)) {
            cin >> U[i] >> V[i] >> a[i];
            ++ot[U[i]],++deg[V[i]];
            add(U[i],V[i]);
            sum+=a[i];
        }
        sum%=MOD;
        INV=fpow(sum,MOD-2);
        inv[1]=1;
        for (ri i(2);i<=k;pd(i)) inv[i]=1ll*(MOD-MOD/i)*inv[MOD%i]%MOD;
        for (ri i(1);i<=m;pd(i)) que.push(i),dp[i]=1ll;
        memcpy(td+1,deg+1,sizeof(int)*n);
        while(!que.empty()) {
            int x=que.front();
            que.pop();
            const int iv=inv[ot[x]];
            for (ri i(first[x]),v;i;i=e[i].nxt) {
                (dp[v=e[i].v]+=dp[x]*iv%MOD)%=MOD;
                --td[v];
                if (!td[v]) que.push(v);
            }
        }
        for (ri i(1);i<=k;pd(i)) {
            const int u=U[i],v=V[i],iv=1ll*a[i]*INV%MOD;
            tdp[u]+=dp[u]*inv[ot[u]-1]%MOD*iv%MOD;
            tdp[v]-=dp[u]*inv[ot[u]-1]%MOD*iv%MOD;
        }
        for (ri i(1);i<=n;pd(i)) {
            dp[i]=tdp[i]%MOD;
            if (i<=m) dp[i]+=1ll,que.push(i);
        }
        while(!que.empty()) {
            int x=que.front();
            que.pop();
            const int iv=inv[ot[x]];
            for (ri i(first[x]),v;i;i=e[i].nxt) {
                (dp[v=e[i].v]+=dp[x]*iv%MOD)%=MOD;
                --deg[v];
                if (!deg[v]) que.push(v);
            }
        }
        for (ri i(n-r+1);i<=n;pd(i)) printf("%lld ",(dp[i]%MOD+MOD)%MOD);
        return 0;
    }
}
int main() {return nanfeng::main();}

标签:ch,NOIP,int,res,++,89,rm,define,MOD
来源: https://www.cnblogs.com/nanfeng-blog/p/15511453.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有