[WC2019] 数树

最新推荐文章于 2021-11-17 14:42:07 发布
最新推荐文章于 2021-11-17 14:42:07 发布
阅读量272 收藏
点赞数
原文链接：http://www.cnblogs.com/Miracevin/p/10698164.html
版权

[WC2019] 数树 

Zhang_RQ题解（本篇仅概述）

 

前言

有进步，只做了半天。。。。

一道具有极强综合性的数数好题！

强大的多合一题目

精确地数学推导和耐心。

有套路又不失心意。

 

融合了：

算法：

prufer序列及其扩展

树形Dp

容斥或者二项式定理

EGF

多项式Exp

 

先要会：

[学习笔记]prufer序列

[学习笔记]多项式指数函数

[学习笔记]生成函数

luoguP4841 城市规划

省选模拟赛第四轮 B——O(n^4)->O(n^3)->O(n^2)

 

然后开始刚题。

就是：

求各种情况下的y^(n-|T1∩T2|)的和

 

 

OP=0

哈希。

OP=1

枚举交集，统计符合的T2树的个数

prufer序列森林版扩展

但是会算重，连回去会多

考虑重组贡献

每个实际的$y^k$会被$k$小的时候计算，$y^k=(y+1-1)^k=∑C(j,i)(y-1)^i$

发现，只要把贡献的$y$成$y-1$，就大功告成啦！

$\Pi szi$可以套路地组合意义成为每个连通块选择一个的方案数，然后就$f[i][0/1]$就O(n)辣

注意每次选择交的边还有贡献

这里，第二个$y^{-1}$运算的时候，要变成$(y^{-1}-1)$

OP=2

还是枚举交集的边

发现和交集大小密切相关，所以枚举交集大小$s$​

然后$gs$设出来，

要枚举边再考虑连通块，不妨枚举连通块考虑贡献的边集

连通块的分配$\frac{n!}{(n-s)!}$,连通块内的连边$a^{a-2}$，连通块之间连边$\Pi a_i *n^{m-2}$

一些东西要平方

把$gs$带回去，然后无关的都提到前面去

把枚举连通块，变成EGF

然后再变成多项式指数函数的形式，大功告成！

 

 

#include<bits/stdc++.h>
#define reg register int
#define il inline
#define fi first
#define se second
#define mk(a,b) make_pair(a,b)
#define numb (ch^'0')
using namespace std;
typedef long long ll;
template<class T>il void rd(T &x){
    char ch;x=0;bool fl=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*10+numb);
    (fl==true)&&(x=-x);
}
template<class T>il void output(T x){if(x/10)output(x/10);putchar(x%10+'0');}
template<class T>il void ot(T x){if(x<0) putchar('-'),x=-x;output(x);putchar(' ');}
template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}

namespace Miracle{
const int N=1e5+5;
const int mod=998244353;
const int G=3;
const int GI=332748118;
int n,Y;
int mul(int x,int y){return (ll)x*y%mod;}
int ad(int x,int y){return x+y>=mod?x+y-mod:x+y;}
il int sub(int x,int y){return ad(x,mod-y);}
int qm(int x,int y){
    int ret=1;
    while(y){
        if(y&1) ret=mul(ret,x);
        x=mul(x,x);
        y>>=1;
    }
    return ret;
}
namespace sol0{
map<pair<int,int>,int>mp;
int cnt;
void main(){
    int x,y;
    for(reg i=1;i<n;++i){
        rd(x);rd(y);if(x>y) swap(x,y);mp[mk(x,y)]=1;
    }
    for(reg i=1;i<n;++i){
        rd(x);rd(y);if(x>y) swap(x,y);cnt+=mp[mk(x,y)];
    }
    printf("%d\n",qm(Y,n-cnt));
}

}
namespace sol1{
ll f[N][2];
int z;
int cos,ivc;
struct node{
    int nxt,to;
}e[2*N];
int hd[N],cnt;
void add(int x,int y){
    e[++cnt].nxt=hd[x];
    e[cnt].to=y;
    hd[x]=cnt;
}
void dfs(int x,int fa){
    f[x][0]=f[x][1]=1;
    for(reg i=hd[x];i;i=e[i].nxt){
        int y=e[i].to;
        if(y==fa) continue;
        dfs(y,x);
        //jiao and no
        ll od0=f[x][0],od1=f[x][1];
        f[x][0]=ad(mul(od0,f[y][1]),mul(od0,mul(f[y][0],cos)));
        f[x][1]=ad(mul(od1,f[y][1]),ad(mul(od0,mul(f[y][1],cos)),mul(od1,mul(f[y][0],cos))));
    }
}
void main(){
    int x,y;
    for(reg i=1;i<n;++i){
        rd(x);rd(y);add(x,y);add(y,x);
    }
    if(Y==1){
        printf("%d\n",qm(n,n-2));
        return;
    }
    z=ad(qm(Y,mod-2),mod-1);
    cos=mul(z,qm(n,mod-2));
    ivc=qm(cos,mod-2);
//    cout<<" cos "<<cos<<" ivc "<<ivc<<endl;
    dfs(1,0);
    ll ans=0;
    ans=mul(mul(qm(n,n-2),qm(Y,n)),f[1][1]);
    cout<<ans;
}
}
namespace sol2{
struct Poly{
    vector<int>f;
    Poly(){f.clear();}
    il int &operator[](const int &x){return f[x];}
    il const int &operator[](const int &x) const {return f[x];}
    il void resize(const int &n){f.resize(n);}
    il int size() const {return f.size();}
    il void cpy(Poly &b){f.resize(b.size());for(reg i=0;i<(int)f.size();++i)f[i]=b[i];}
    il void rev(){reverse(f.begin(),f.end());}
    il void clear(){f.clear();}
    il void read(const int &n){f.resize(n);for(reg i=0;i<n;++i)rd(f[i]);}
    il void out() const {for(reg i=0;i<(int)f.size();++i)ot(f[i]);putchar('\n');}
}R;
il int init(const int &n){int m;for(m=1;m<n;m<<=1);return m;}
il void rev(Poly &f){
    int lp=f.size();
    if(R.size()!=f.size()) {
        R.resize(f.size());
        for(reg i=0;i<lp;++i){
            R[i]=(R[i>>1]>>1)|((i&1)?lp>>1:0);
        }
    }
    for(reg i=0;i<lp;++i){
        if(i<R[i]) swap(f[i],f[R[i]]);
    }
}
il void NTT(Poly &f,int c){
    int n=f.size();rev(f);
    for(reg p=2;p<=n;p<<=1){
        int gen=(c==1)?qm(G,(mod-1)/p):qm(GI,(mod-1)/p);
        for(reg l=0;l<n;l+=p){
            int buf=1;
            for(reg k=l;k<l+p/2;++k){
                int tmp=mul(f[k+p/2],buf);
                f[k+p/2]=sub(f[k],tmp);
                f[k]=ad(f[k],tmp);
                buf=mul(buf,gen);
            }
        }
    }
    if(c==-1){
        int iv=qm(n,mod-2);for(reg i=0;i<n;++i) f[i]=mul(f[i],iv);
    }
}
il Poly Inv(const Poly &f,int n){
    if(n==1){
        Poly g;g.resize(1);g[0]=qm(f[0],mod-2);return g;
    }
    Poly g=Inv(f,(n+1)>>1),t;
    int m=init(n*2);
    t.resize(m);
    for(reg i=0;i<n;++i)t[i]=f[i];
    g.resize(m);
    NTT(g,1);NTT(t,1);
    for(reg i=0;i<m;++i)g[i]=mul(sub(2,mul(g[i],t[i])),g[i]);
    NTT(g,-1);g.resize(n);
    return g;
}
il void operator *=(Poly &f,Poly g){
    int st=f.size()+g.size()-1;
    int len=init(f.size()+g.size()-1);f.resize(len);g.resize(len);
    NTT(f,1);NTT(g,1);for(reg i=0;i<len;++i) f[i]=mul(f[i],g[i]);
    NTT(f,-1);
    f.resize(st);
}
il void operator *=(Poly &f,const int &c){for(reg i=0;i<f.size();++i) f[i]=mul(f[i],c);}
il Poly operator *(Poly f,const Poly &g){f*=g;return f;}
il Poly operator *(Poly f,const int &c){for(reg i=0;i<f.size();++i) f[i]=mul(f[i],c);return f;}
il void operator +=(Poly &f,const Poly &g){for(reg i=0;i<f.size();++i) f[i]=ad(f[i],g[i]);}
il void operator +=(Poly &f,const int &c){f[0]=ad(f[0],c);}
il Poly operator +(Poly f,const Poly &g){for(reg i=0;i<f.size();++i) f[i]=ad(f[i],g[i]);return f;}
il Poly operator +(Poly f,const int &c){f[0]=ad(f[0],c);return f;}
il void operator -=(Poly &f,const Poly &g){for(reg i=0;i<f.size();++i) f[i]=sub(f[i],g[i]);}
il void operator -=(Poly &f,const int &c){f[0]=sub(f[0],c);}
il Poly operator -(Poly f,const Poly &g){for(reg i=0;i<f.size();++i) f[i]=sub(f[i],g[i]);return f;}
il Poly operator -(Poly f,const int &c){f[0]=sub(f[0],c);return f;}
il Poly operator ~(const Poly &f){return Inv(f,f.size());}
il Poly operator /(Poly f,Poly g){int len=f.size()-g.size()+1;f.rev();g.rev();g.resize(len);f=f*(~g);f.resize(len);f.rev();return f;}
il Poly operator %(Poly f,Poly g){Poly s=f/g;f=f-g*s;f.resize(g.size()-1);return f;}
il Poly Inter(Poly f){int st=f.size();f.resize(st+1);for(reg i=st;i>=1;--i){f[i]=mul(f[i-1],qm(i,mod-2));}f[0]=0;return f;}
il Poly Diff(Poly f){int st=f.size();for(reg i=0;i<st-1;++i) f[i]=mul(f[i+1],(i+1));f.resize(st-1);return f;}
il Poly Ln(const Poly &f){Poly g=Diff(f),h=(~f);g=g*h;g.resize(f.size()-1);return Inter(g);}
il Poly Exp(const Poly &f,int n){
    if(n==1){
        Poly g;g.resize(1);g[0]=1;
        return g;
    }
    Poly g=Exp(f,(n+1)>>1);
    g.resize(n);
    g=g*(((Ln(g)*(mod-1))+1)+f);
    g.resize(n);
    return g;
}
il Poly Exp(const Poly &f){
    return Exp(f,f.size());
}
int jie[N],inv[N];
void main(){
    if(Y==1){
        printf("%d\n",mul(qm(n,n-2),qm(n,n-2)));
        return;
    }
    int z=ad(qm(Y,mod-2),mod-1);
    Poly g;
    g.resize(n+1);
    
    int ivz=qm(z,mod-2);
    jie[0]=1;
    for(reg i=1;i<=n;++i) jie[i]=mul(jie[i-1],i);
    inv[n]=qm(jie[n],mod-2);
    for(reg i=n-1;i>=0;--i) inv[i]=mul(inv[i+1],i+1);
    
    for(reg j=1;j<=n;++j){
        g[j]=mul(mul(mul(mul(n,n),ivz),inv[j]),qm(j,j));
    }
    g=Exp(g);
    ll ans=mul(mul(mul(mul(qm(Y,n),jie[n]),qm(qm(n,mod-2),4)),qm(z,n)),g[n]);
    cout<<ans;
}

}
int main(){
    int op;
    rd(n);rd(Y);rd(op);
    if(op==0) sol0::main();
    else if(op==1) sol1::main();
    else sol2::main();
    return 0;
}

}
signed main(){
    Miracle::main();
    return 0;
}

/*
   Author: *Miracle*
   Date: 2019/4/12 15:09:42
*/

 

转载于:https://www.cnblogs.com/Miracevin/p/10698164.html

weixin_34146410
关注
【LOJ2983】「WC2019」数树
cz_xuyixuan的博客
03-31 655
【题目链接】 点击打开链接 【思路要点】 op=0op=0op=0 ，算一算两棵树的公共边数即可。 时间复杂度 O(N)O(N)O(N) 或 O(NLogN)O(NLogN)O(NLogN) 。 op=1op=1op=1 ，考虑枚举蓝树上的一个边集 SSS ，强制红树上同样存在这些边，计算将剩余 N−∣S∣N-|S|N−∣S∣ 个联通块连成一棵树的方案数，更新答案。 上述算法中，枚举蓝...
【清华集训2017】生成树计数 &【WC2019】数树
Joker's blog
02-03 1385
扩展prufer序列到N结点连成m个连通块的情形。 结论：生成树个数为 N^(m-2) \prod ai。 我们来表演一下用这个结论秒切WC题
[WC2019]数树
时光真疯狂, 我一路执迷于匆忙.
02-27 343
Description 有两棵n个点的树，T1和T2，每个点可以填一个[1,y]的颜色 若两棵树有一条边重合这条边两个端点的颜色必须相同 有三种问题： op=0，给出T1和T2问答案 op=1，给出T1问所有T2的答案和 op=2，给出n问所有的T1和T2的答案和 n&lt;=100000 Solution op=0大家都会 op=1的话，考虑算恰好有i条重边的y−iy^{-i}y−i的和 考虑到...
WC2019 数树
weixin_30908941的博客
02-25 115
多谢yn大佬的指点。 op=0 挺简单的。 op=1 设\(T1,T2\)为两棵树 考虑我们知道\(T1\),不知道\(T2\) 但是答案只是多了个\(\sum\) \[ans= \sum_{T2} y^{|T1 \& T2|} \] 考虑重合了\(m\)条边 \[ans= \sum_{T2} y^{n-m} \] \[ans= \sum_{m=0}^n y^{n-m} f(m)\] \(...
[WC2019] 数树 容斥原理+矩阵树定理+树形Dp+计数Dp+生成函数优化Dp+多项式求Exp
maxtir的博客
06-20 573
[WC2019] 数树 题目传送门 分析 最近老是在肝一些神仙生成函数题。。。哎，肝败吓疯。其实luogu题解里面的那篇已经很详细了，这篇题解纯属个人整理，建议是到到luogu题解去看。 题目大意：告诉你有俩棵有标号无根树，如果某两个节点共用了某条边，那么这两个点的权值必须相同，点权范围在[1,y][1,y][1,y]内，有三个任务，求在给定2,1,0棵树的情况下构造树和点权的方案数。 Task1...
[XXSY] 构树（prufer序列，树上连通块DP）
Edward The Bunny 的博客
11-17 589
传送门 CayleyCayleyCayley公式：一个完全图有nn−2n^{n-2}nn−2棵无根生成树（可用prufer序列证明） 扩展CayleyCayleyCayley公式：被确定边分为大小为a1,a2,⋯ ,ama_1,a_2,\cdots, a_ma1​,a2​,⋯,am​的连通块，则有nm−2∏i=1main^{m-2}\prod\limits_{i=1}^{m}{a_i}nm−2i=1∏m​ai​种生成树 我们可以强制生成树中的边是否与原树中的边相同。 设dp[u][i][j]dp[u][i]
THUWC2019 场外游记
mayaohua
01-31 1481
又开始了冬令营时间。 由于今年THUWC和NOIWC都在我校举办，于是并不需要出去住。但我也不打算住在学校，每天晚上都滚回家。 Day1 早上来到学校，貌似到的很晚（？）。报到完后领了胸牌和衣服，就去吃饭了，饭菜比平常的好多了。 吃完饭回宿舍休息一下，被抓去合影，自信不穿发的衣服。旁边有一个yww红太阳，后面传来“一时膜yww一时爽，一直膜yww一直爽”的声音。 然后就比赛了，毕竟是第四次来，作为...
WC2019数树
Fire Dancer
02-21 871
首先由于我们很难直接求解边集之交为Ｓ的树的个数，所以考虑容斥，转化成至少交Ｓ，这里需要推一波式子，然后利用组合意义，转化成一个连通块内给一个点染色，就可以O(n)　DP了。 这类题能够转化成包含的好算一定要尝试着去凑一下，很有可能可以出奇迹。 然后再推一推就可以推出一个多项式exp的式子啦，感觉还是很妙妙的QAQ。 prufer序列那一部分可以用生成函数的exp来推。 ...
【WC2019】数树【计数】【DP】【多项式】
Never give in.
04-20 265
Description 此题含有三个子问题 问题1： 给出n个点的两棵树，记m为同时在两棵树中的边的个数，求ymy^mym 问题2： 给出n个点的一棵树，另外一棵树任意生成，求所有方案总的ymy^mym的和 问题3： 两棵树均任意生成，求所有方案总的ymy^mym的和 Solution 留坑待填 Code #include <bits/stdc++.h> #define fo(i,a,...
洛谷P5206/loj2983 [WC2019]数树
litble的成(tui)长(fei)史
02-13 464
今年年初，我将继续扮演爆零士，Q1Q2双爆零，弘扬爆零文化，希望大家多多资磁。 这位大佬写的题解真是精妙啊-&amp;amp;amp;amp;gt;here 问题0 记mmm为颜色种类，T1T_1T1​为树1的边集，T2T_2T2​为树2的边集，那么答案就是mn−∣T1∩T2∣m^{n-|T_1 \cap T_2|}mn−∣T1​∩T2​∣，用个map哈希一下即可。 问题1 基本推导 T1T_1T1​已经给定了，T2T_2T2​...
[LOJ#2983][WC2019]数树
女装的你，如此好看！
03-08 404
Address 洛谷 P5206 LOJ #2983 Solution 这是一道计数 (shen) 好 (xian) 题，综合性很强 ，虽然是三合一 鉴于此题用到的算法较多，我就不在标题里列出了，此题用到的算法及技巧如下 容斥原理 prufer 序列计数 组合数学 树形 DP 多项式 exp Task 1 ： op=0op=0op=0 首先分析下问题，很容易得出，如果设 T0T_0T0​...
「WC 2019」数树
weixin_30409849的博客
04-17 206
「WC 2019」数树 一道涨姿势的EGF好题，官方题解我并没有完全看懂，尝试用指数型生成函数和组合意义的角度推了一波。考场上只得了 44 分也暴露了我在数数的一些基本套路上的不足，后面的 \(\exp\) 是真的神仙，做不出来当然很正常，而且我当时也不怎么会多项式。 Task0 考虑公共边组成 \(k\) 个联通块，答案就是 \(y^k\) ，并查集维护一下即可，复杂度 \(\mathcal O...
LOJ#2983. 「WC2019」数树
Cyhlnj
02-01 262
传送门 抄题解 Task0Task0Task0，随便做一下，设 cntcntcnt 为相同的边的个数，输出 yn−cnty^{n-cnt}yn−cnt Task1Task1Task1，给定其中一棵树 设初始答案为 yny^nyn，首先可以发现，每有一条边和给定的树相同就会使得答案除去 yyy 那么可以利用矩阵树定理，已经有的边权值为 y−1y^{-1}y−1，其它的连成完全图，权值为 111 求解...
数树
02-19 396
本来想叫 WC2019 数树 的 后来发现自己还不怎么会怎么数树... 那就来数数树吧 Matrix-Tree 定理 令 $G$ 是一个无向图，$D$ 为 $G$ 的度数矩阵，即 $D[i][i] = [i的度数]$，$A$ 为 $G$ 的邻接矩阵，即 $A[i][j] = [i,j之间边的数量]$ 定义 $G$ 的 基尔霍夫矩阵 $K$ 为 $D-A$ 则 $K$ 随...
【WC2019】数树（容斥原理）（生成函数）（树形DP）（多项式Exp）（数数神题）
zxyoi_dreamer的博客（不定期诈尸）
10-29 288
传送门 没弄明白前觉得毒瘤，弄明白之后，这TM就是数数神题啊。 然而我这道题将近一半的代码都在写多项式全家桶。。。 这道题的推导过程确实有点繁琐，但是没有办法啊，毕竟性质就是这么复杂啊。 白云和白兔（laofu出题用的NPC），老虎和蒜头（whzzt出题用的NPC），真心有毒真的牛逼啊。。。 题解： 首先简单分析一下题目的那个限制：有公共路径的点颜色必须一样。 很容易发现其实就是两棵树里面的公共...
「WC2019」数树（生成函数）（树形DP）（容斥）（多项式exp）
FSYo 的博客
03-08 336
LOJLOJLOJ 传送门 转换一下题意得到：对于两棵树 T1,T2T1,T2T1,T2，边集是 E1,E2E1,E2E1,E2，那么贡献就是 yn−∣E1∩E2∣y^{n-|E1\cap E2|}yn−∣E1∩E2∣ 第一问：直接求出并即可 第二问：T1T1T1 确定，不妨来枚举 T2T2T2 令 F(S)=∑T2[T1∩T2=S]F(S)=\sum_{T2} [T1\cap T2=...
wc 命令字节数怎么算
最新发布
05-24
wc 命令用于统计文件中的行数、字数和字符数等信息。其中，字节数是指文件中的所有字节总数，包括空格、制表符、换行符等。 在使用 wc 命令时，字节数可以通过参数 "-c" 来指定统计。例如，假设要统计文件 "example.txt" 中的字节数，可以执行以下命令： ``` wc -c example.txt ``` 运行结果会显示文件中的字节数。需要注意的是，这个字节数并不一定等于文件大小，因为文件大小可能包含一些元数据（如文件名、创建时间等）或者磁盘上的块大小等。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值