【纪中集训2019.3.12】Mas的仙人掌

题意:

? 给出一棵\(n\)个点的树,需要加\(m\)条边,每条边脱落的概率为\(p_{i}\) ,求加入的边在最后形成图中仅在一个简单环上的边数的期望;

题解:

  • 考虑每一条边的贡献是\((1-p_{i})*\Pi_{j}p_{j}(j!=i)\),这里\(j\)和\(i\)不能同时加入;
  • 一条加入的边可以看成一条树上路径 ,即求所有和路径\(i\)相交的路径\(j\)的\(p_{j}\)的乘积;
  • 将一条树上的链\((u,v)\)拆成两条\((u,lca)\)和\((v,lca)\);
  • 这样会算重复\(i\)和\(j\)的两条链都相交的情况,但是这样\((u,lca)\)上\(lca\)的儿子是唯一的,\(v\)同理,\(hash\)去掉多算的部分;
  • 考虑一条路径\((u,v) dep[u]<dep[v]\),考虑一个点的子树里向上有长度为\(2\)的链的个数,简单容斥预处理出在\(v\)的子树里但不在\((u,v)\)上的点对答案的贡献;
  • 在\((u,v)\)上的点直接每条链在最下端打标记即可;
    #include<bits/stdc++.h>
    #define ll long long
    #define mod 998244353
    using namespace std;
    const int N=1000010;
    int n,m,o=1,hd[N],a[N],b[N],ia[N],ans[N];
    int fa[N][21],bin[21],dep[N],cnt,tot;
    int s1[N],s2[N],s3[N];
    ll s4[N];
    struct Edge{int v,nt;}E[N<<1];
    struct data{
      int u,v,w,id;
    }A[N],B[N<<1];
    int pw(int x,int y){
      int re=1;
      while(y){
          if(y&1)re=(ll)re*x%mod;
          y>>=1;x=(ll)x*x%mod;
      }
      return re;
    }//
    void adde(int u,int v){
      E[o]=(Edge){v,hd[u]};hd[u]=o++;
      E[o]=(Edge){u,hd[v]};hd[v]=o++;
    }//
    char gc(){
      static char*p1,*p2,s[1000000];
      if(p1==p2)p2=(p1=s)+fread(s,1,1000000,stdin);
      return(p1==p2)?EOF:*p1++;
    }//
    int rd(){
      int x=0;char c=gc();
      while(c<'0'||c>'9')c=gc();
      while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+c-'0',c=gc();
      return x;
    }//
    const int sz=1234651;
    struct HASH{
      int o,U[N],V[N],w[N],hd[sz],nt[N];
      int ask(int u,int v){
          if(u>v)swap(u,v);
          int x = ((ll)u*mod+v)%sz;
          for(int i=hd[x];i;i=nt[i]){
              if(U[i]==u&&V[i]==v)return w[i];
          }
          return 1;
      }
      void upd(int u,int v,int y){
          if(u>v)swap(u,v);
          int x = ((ll)u*mod+v)%sz;
          for(int i=hd[x];i;i=nt[i]){
              if(U[i]==u&&V[i]==v){w[i]=(ll)w[i]*y%mod;return;}
          }
          nt[++o]=hd[x];hd[x]=o;U[o]=u,V[o]=v;w[o]=y;
      }
    }H;//
    void dfs1(int u,int F){
      dep[u]=dep[fa[u][0]=F]+1;
      for(int i=1;bin[i]<dep[u];++i)fa[u][i]=fa[fa[u][i-1]][i-1];
      for(int i=hd[u];i;i=E[i].nt){
          int v=E[i].v;
          if(v==F)continue;
          dfs1(v,u);
      }
    }//
    int go(int u,int d){
      for(int i=0;i<20&&d;++i)if(d&bin[i])d^=bin[i],u=fa[u][i];
      return u;
    }//
    int lca(int u,int v){
      if(dep[u]<dep[v])swap(u,v);
      u=go(u,dep[u]-dep[v]);
      if(u==v)return u;
      for(int i=19;~i;--i)if(fa[u][i]!=fa[v][i])u=fa[u][i],v=fa[v][i];
      return fa[u][0];
    }//
    void dfs2(int u,int F){
      for(int i=hd[u];i;i=E[i].nt){
          int v=E[i].v;
          if(v==F)continue;
          dfs2(v,u);
          s2[u]=(ll)s2[u]*s2[v]%mod;
          s3[u]=(ll)s3[u]*s2[v]%mod;
          s4[u]=(s4[u]+s4[v]);
      }
    }//
    void dfs3(int u,int F){
      for(int i=hd[u];i;i=E[i].nt){
          int v=E[i].v;
          if(v==F)continue;
          s1[v]=(ll)s1[u]*s1[v]%mod;
          s2[v]=(ll)s2[u]*s2[v]%mod;
          s3[v]=(ll)s3[u]*s3[v]%mod;
          s4[v]=(s4[u]+s4[v]);
          dfs3(v,u);
      }
    }//
    inline int cal1(int u,int v){return (ll)s1[u]*pw(s1[v],mod-2)%mod;}
    inline int cal2(int u,int v){return (ll)s2[u]*pw(s2[v],mod-2)%mod;}
    inline int cal3(int u,int v){return (ll)s3[u]*pw(s3[v],mod-2)%mod;}
    inline ll cal4(int u,int v){return s4[u]-s4[v];}
    //
    int main(){
      freopen("cactus.in","r",stdin);
      freopen("cactus.out","w",stdout);
      n=rd();m=rd();
      for(int i=bin[0]=1;i<=20;++i)bin[i]=bin[i-1]<<1;
      for(int i=1;i<n;++i)adde(rd(),rd());
      dfs1(1,0);
      for(int i=1;i<=max(n,m);++i){ans[i]=s1[i]=s2[i]=s3[i]=1;}
      for(int i=1,u,v,w,p,q;i<=m;++i){
          u=rd();v=rd();w=lca(u,v);
          a[i]=rd();ia[i]=pw(a[i],mod-2);
          b[i]=(1+mod-a[i])%mod;
          if(!a[i]){
              s4[u]++;s4[v]++;s4[w]-=2;
              if(u!=w){
                  p=go(u,dep[u]-dep[w]-1);
                  B[++cnt]=(data){u,w,p,i};
              }
              if(v!=w){
                  q=go(v,dep[v]-dep[w]-1);
                  B[++cnt]=(data){v,w,q,i};
              }
              if(u!=w&&v!=w){
                  A[++tot]=(data){p,q,0,i};
              }
              continue;
          }//
          if(u!=w){
              p=go(u,dep[u]-dep[w]-1);
              B[++cnt]=(data){u,w,p,i};
              s1[u]=(ll)s1[u]*a[i]%mod;
              s2[u]=(ll)s2[u]*a[i]%mod;
              s2[p]=(ll)s2[p]*ia[i]%mod;
          }
          if(v!=w){
              q=go(v,dep[v]-dep[w]-1);
              B[++cnt]=(data){v,w,q,i};
              s1[v]=(ll)s1[v]*a[i]%mod;
              s2[v]=(ll)s2[v]*a[i]%mod;
              s2[q]=(ll)s2[q]*ia[i]%mod;
          }
          if(u!=w&&v!=w){
              A[++tot]=(data){p,q,0,i};
              H.upd(p,q,ia[i]);
          }
      }
      dfs2(1,0);
      dfs3(1,0);
      for(int i=1;i<=tot;++i){
          int u=A[i].u,v=A[i].v;
          ans[A[i].id]=(ll)ans[A[i].id]*H.ask(u,v)%mod;
      }
      for(int i=1;i<=cnt;++i){
          int u=B[i].u,v=B[i].v,w=B[i].w,now=1;
          now=cal3(u,v);
          now=(ll)now*pw(cal2(u,w),mod-2)%mod;
          now=(ll)now*cal1(u,v)%mod;
          ans[B[i].id]=(ll)ans[B[i].id]*now%mod;
          int t = cal4(u,v);
          if(!a[B[i].id])t-=dep[u]-dep[v];
          if(t)ans[B[i].id]=0;
      }
      int Ans=0;
      for(int i=1;i<=m;++i){
          ans[i] = (ll)ans[i]*b[i]%mod;
          if(a[i])ans[i] = (ll)ans[i]*pw(a[i],mod-2)%mod;
          Ans=(Ans+ans[i])%mod;
      }
      cout<<Ans<<endl;
      return 0;
    }

原文地址:https://www.cnblogs.com/Paul-Guderian/p/10519601.html

时间: 2024-11-06 07:14:08

【纪中集训2019.3.12】Mas的仙人掌的相关文章

【纪中集训2019.3.29】整除分块

题目 描述 ? 本题的背景是整除分块: ? 定义一个数列$a_{n,i} ? = ?\lfloor \frac{n}{i} \rfloor $ ; ? 求$\sum_{i=l}^{r} mex(a_n) $ ; ? 其中\(mex\)表示序列中最小的没有出现过的自然数: ? 答案对\(998244353\)取模 : 范围 ? \(1 \le T \le 65536 \ , \ 1 \le l ,r \le 10^{36}\) : ? 评测系统支持使用 $ _ _ int218 $ ,但是不能直接

【纪中集训2019.08.20】【JZOJ6310】Global warming

题目链接 题意: 给出一个长度为$n$的序列$\{a_n\}$. 已知一个正整数$x$,你有一次机会指定区间$[l,r]$,令$\forall i\in [l,r],\;a_i=a_i+d\,(|d|\le x)$. 求最大化的最长上升子序列的长度. $1 \le n \le 2 \times 10^5 , \quad 1 \le a_i,x \le 10^9$且均为正整数. 分析: 可以很容易发现三个性质: 性质一:抬升$[l,r]$不优于抬升$[l,n]$,降低$[l,r]$不优于降低$[1

【纪中集训2019.3.20】河

题目 描述 给出\(n\)条河流,每条河流是形式为\(k_{i}x+b_{i}\)的一次函数且只有\(x\)轴正半轴的部分: 河流的污染部分和另外一条河流的干净部分交汇,干净的部分会被污染: 有若干个工厂要建在\(b_{i}\)处,问有多少种方案使得所有的河流在无穷远处被污染: 答案对\(1e9+7\)取模 范围 $1 \le N \le 5 \times 10^5 ? ?, ??|k_{i}|,|b_{i}| \le 10^9 $ 保证\(b_{i}\)互不相同: 题解 需要先找到可以被一条河

【纪中集训2019.08.21】【JZOJ6315】数字

题目链接 题意: 设$s(i)$为将$1\sim i$看做字符串后依次连接形成的串.给定正整数$n$,求最小的$i$使得$n$是$s(i)$的字串.$T$组数据. $n\le 10^{17}, \; t\le 10^4$ 分析: 不能模拟$s(i)$的组成过程来找答案,时间不能承受. 也不能预处理$s(k)$,空间不能承受. 那就只能在$n$上找答案. 以下把数字当成字面量来讨论,更方便. 同时,这里讨论的前缀和后缀不包括本身. 思考一下,答案分为三种: 1.$ans=n$ 2.$n$由$ans

纪中集训2019.11.06

A.困难的图论 题目链接 题意: 给出由$n$个点和$m$条边构成的无向连通图,要求选出一些边.一条边被选中当且仅当它恰好被一个简单环经过. 一个环被称为简单环,当且仅当这个环上的所有点都只在这个环中被经过了一次. 输出这些边的编号的异或和.边从$1$开始编号. $1\le n\le 10^6,\;1\le m\le min\{10^6,n\times (n+1) /2\}$. 分析: 注意:一个图可能有很多简单环,题目要求选中所有简单环上的边.考场上我就是这一点搞错了.(其实搞对了也很可能做不

纪中集训2019.11.07

A.极好的问题 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 B.背包问题 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 C.子树问题 题目链接 题意: 咕 分析: 咕 小结: 咕 当天总结: 咕 原文地址:https://www.cnblogs.com/Hansue/p/11811824.html

纪中集训2019.11.09

A.搞笑的代码 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 B.道路维护 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 C.字符串查询 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 当天总结: 咕 原文地址:https://www.cnblogs.com/Hansue/p/11826647.html

纪中集训2019.11.08

A.三角形计数 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 B.Isaac 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 C.字符消除 题目链接 题意: 咕 分析: 咕 实现: 咕 小结: 咕 当天总结: 咕 原文地址:https://www.cnblogs.com/Hansue/p/11826709.html

木兮的纪中集训感想

嗯 第一次写博客... 就记录一下我这个OI萌新在纪中集训的感受吧 ## 学校 对于纪中,我这几天下来还是非常认可这所学校的:全国最美中学名不虚传(宿舍除外),校园环境非常好,虽然建校将近百年,但维护工作做得非常到位,看不出太多时间的痕迹. 训练 抛开学校环境不讲,这里的教学也是非常棒的,我来这里之前一直觉得集训应该是一件非常累的事,但事实并没有我想象的那么坏.先放一个训练的时间表 8:00 ~ 11:20 打模拟 14:00 ~ 17:30 讲题+自己改程序 19:00 ~ 21:50 自习