Fantasia (Tarjan+树形DP)

Time Limit: 1000 ms   Memory Limit: 256 MB

Description

  给定一张N个点、M条边的无向图 $G$ 。每个点有个权值Wi。

  我们定义 $G_i$ 为图 $G$ 中删除第 $i$ 号顶点后的图。我们想计算 $G_1, G_2, ..., G_n$ 这N张图的权值。

  对于任意一张图 $G$ ,它的权值是这样定义的:

  1. 如果 $G$ 是联通图,那么 $G$ 的权值为 $G$ 中所有顶点权值的乘积。

  2. 如果 $G$ 是非联通图,那么 $G$ 的权值为 $G$ 中所有联通块的权值之和。

  $G$ 中的一个联通块指的是 $G$ 的一个子图,并且这个子图中的点两两相连(包括直接连接或者间接连接),并且不存在子图外的点使得子图内的点能与子图外的点相连。

Input

  第一行包含两个整数 $n$ 和 $m$ $(2 \le n \le 10^5, 1 \le m \le 2 \times 10^5)$ ,分别表示点数和边数。

  第二行包含 $n$ 个整数 $w_1, w_2, ..., w_n$ $(1 \le w_i \le 10^9)$, 表示每个顶点的权值。

  接下来 m 行,每行两个整数 $x_i$ 和 $y_i$ $(1 \le x_i, y_i \le n, x_i \ne y_i)$, 表示一条无向边。

  输出只有一个整数: $S = (\sum\limits_{i=1}^{n}i\cdot z_i) \text{ mod } (10^9 + 7)$, 其中 $z_i$ 是图 $G_i$ 的权值。

Sample Input

Sample Output

10 3
3 3 3
2 3 3
2 3 1
3 1 1
3 1 2
1 3 1
1 1 2
1 2 2
1 3 2
1 2 1
3
1
3
0
1
0
1
0
0
1

Hint

  【数据范围及约定】

  子任务1(5分): $n \leq 10, m \leq 20$

  子任务2(10分): $n \leq 1000, m \leq 2000$

  子任务3(20分): 该图恰为一棵树,$m = n-1$

  子任务4(20分): 该图为一幅联通图

  子任务5(45分): 我们会拿最强的数据来评测你的程序(mmp)

  对于所有数据,$2 \le n \le 10^5, 1 \le m \le 2 \times 10^5$


题解

  没有什么能阻挡我把Tarjan打残。

  题目涉及到删点操作。

  如果删的点$u$是一个非割顶,那么它的消失貌似对这个联通块整体没有太大的影响,仅仅是该当前联通块的权值$val$除去$u$的权值$w_u$。

  如果删的点$u$是一个割顶,那么它会将这个联通块分成若干部分,具体就是在Tarjan的缩点树上,把子树全部断开,把父亲也断开。问题来了,割顶这个东西很烦怎么处理?

 

转树

  割顶出现了!它可以同时处于多个点双内,mmp

  对于每个点双,我们暂且新建一个代表点,将点双内的所有点连向这个代表点。这样,一个割顶可以被连接到多个点双的代表点,同时整个图转成了树的形态。

  

  那么断开一个割顶$u$会影响到哪些区块,就一目了然了,即这种树上,$u$的所有子树和父亲那一头的部分。

  维护

  $$f_u=\prod\limits_{v\in 以u为根的树}w[v]\\g_u=\sum\limits_{v是u的子树}f[v]$$

  则删去一个割顶$u$,对所在联通块权值$val$的影响即为:

  $$val=\frac{val}{f_u}+g_u$$

    即父亲那一头的权值+所有子树的权值和

小细节与特判

  1.处理删去割顶的时候(即上面的最后一个公式),$\frac{val}{f_u}$希望得到的是父亲那一头的权值,但如果$u$是树的根,这玩意弄出来却是1,而不是我们希望的0(坑爹),所以记录一下我们要处理的割顶是不是一个树的根,特判一下。

    2.Tarjan深搜的起始点要记为割顶。



 1 #include <cstdio>
 2 #define min(a,b) (a<b?a:b)
 3 using namespace std;
 4 typedef long long ll;
 5 const ll N=200010,Mod=1e9+7;
 6 int n,m,h1[N],h2[N*2],tot;
 7 int col[N],colcnt,st[N],top,bcnt,head[N];
 8 ll info[N],sumup,ans,f[N*2],g[N*2],w[N*2];
 9 int dfn[N],low[N],ins[N],tmcnt;
10 bool cut[N];
11 struct Edge{int v,next;}G[N*6];
12 inline void addEdge(int u,int v,int *h){
13     G[++tot].v=v; G[tot].next=h[u]; h[u]=tot;
14 }
15 void tarjan(int u,int fa){
16     st[++top]=u;
17     ins[u]=1;
18     dfn[u]=low[u]=++tmcnt;
19     col[u]=colcnt;
20     info[col[u]]=(info[col[u]]*w[u])%Mod;
21     for(int i=h1[u],v,ccnt=0;i;i=G[i].next)
22     if((v=G[i].v)!=fa){
23         if(!ins[v]){
24             ccnt++;
25             tarjan(v,u);
26             low[u]=min(low[u],low[v]);
27             if((!fa&&ccnt>1)||(fa&&dfn[u]<=low[v]))
28                 cut[u]=1;
29             if(dfn[u]<=low[v]){
30                 w[(++bcnt)+n]=1;
31                 do{
32                     addEdge(st[top],bcnt+n,h2);
33                     addEdge(bcnt+n,st[top],h2);
34                     top--;
35                 }while(st[top+1]!=v);
36                 addEdge(u,bcnt+n,h2);
37                 addEdge(bcnt+n,u,h2);
38             }
39         }
40         else if(ins[v]==1)
41             low[u]=min(low[u],dfn[v]);
42     }
43     ins[u]=2;
44 }
45 void dfs(int u,int fa){
46     f[u]=w[u]; g[u]=0;
47     for(int i=h2[u],v;i;i=G[i].next)
48         if((v=G[i].v)!=fa){
49             dfs(v,u);
50             f[u]=(f[u]*f[v])%Mod;
51             g[u]=(g[u]+f[v])%Mod;
52         }
53 }
54 ll ksm(ll bas,ll tm){
55     if(tm==0) return 1;
56     ll ret=ksm(bas,tm/2);
57     ret=(ret*ret)%Mod;
58     return ((tm&1)?ret*bas:ret)%Mod;
59 }
60 ll inv(int x){return ksm(x,Mod-2);}
61 int main(){
62     scanf("%d%d",&n,&m);
63     for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
64     for(int i=1,u,v;i<=m;i++){
65         scanf("%d%d",&u,&v);
66         addEdge(u,v,h1); addEdge(v,u,h1);
67     }
68     for(int i=1;i<=n;i++)
69         if(!dfn[i]){
70             info[++colcnt]=1;
71             tarjan(i,0);
72             cut[i]=1;
73             sumup=(sumup+info[colcnt])%Mod;
74             head[colcnt]=i;
75             dfs(i,0);
76         }
77     for(ll i=1,k;i<=n;i++){
78         int c=col[i];
79         if(!cut[i])
80             k=(sumup+Mod*2-info[c]+(info[c]*inv(w[i]))%Mod)%Mod;
81         else{
82             if(head[c]!=i) k=(sumup+Mod*2-info[c]+(info[c]*inv((f[i])%Mod)%Mod)%Mod+g[i])%Mod;
83             else k=(sumup+Mod*2-info[c]+g[i])%Mod;
84         }
85         ans=(ans+(i*k)%Mod)%Mod;
86     }
87     printf("%lld\n",ans);
88     return 0;
89 }

奇妙代码

时间: 2024-08-07 08:44:39

Fantasia (Tarjan+树形DP)的相关文章

hdu2422考研路茫茫——空调教室 tarjan+树形dp

//给一个无向图,其每个顶点都有权值,求去掉一条边,将这个图分为两部分 //问这两部分的所有顶点和的绝对值的最小值 //用tarjan缩点 , 缩点后为一棵树 //然后用树形dp求出其最小的绝对值 // ans = min(ans , (int)(abs((double)(sum - 2*dp[v])))) ; //其中dp[u] 表示以u点为根节点的子树的学生数 #include<cstdio> #include<cstring> #include<iostream>

bzoj 2427: [HAOI2010]软件安装【tarjan+树形dp】

一眼最大权闭合子图,然后开始构图,画了画之后发现我其实是个智障网络流满足不了m,于是发现正确的打开方式应该是一眼树上dp 然后仔细看了看性质,发现把依赖关系建成图之后是个奇环森林,这个显然不能直接dp 发现这个环要选的话只能选整个环,所以tarjan缩一下点,然后再跑树上背包就行了 #include<iostream> #include<cstdio> #include<vector> #include<cstring> using namespace st

hdu2242(树形dp+tarjan+缩点)

hdu2242 http://acm.hdu.edu.cn/showproblem.php?pid=2242 给定n,m表示n个点,m条边 每个点有个权值 问我们删除两某条边(割边)后将图分为两个部分,要使得两个部分的权值之差最小 这题的弱化版本是在一棵树上删除某条边后后将图分为两个部分,要使得两个部分的权值之差最小.是用树形dp来做的 但是这道题目是个图,但是我们可以转化为树,即将图中的边连通分量求出来,然后缩成一个点,建出一个新的树图,那么就可以用树形dp来求解题目了. 1 #include

[Bzoj 2427] [HAOI2010] 软件安装 tarjan缩点+树形DP

题目描述 现在我们的手头有N个软件,对于一个软件i,它要占用Wi的磁盘空间,它的价值为Vi.我们希望从中选择一 些软件安装到一台磁盘容量为M计算机上,使得这些软件的价值尽可能大(即Vi的和最大). 但是现在有个问题:软件之间存在依赖关系,即软件i只有在安装了软件j(包括软件j的直接或间接依赖)的情况下才能正确工作(软件i依赖软件j).幸运的 是,一个软件最多依赖另外一个软件.如果一个软件不能正常工作,那么它能够发挥的作用为0. 我们现在知道了软件之间的依赖关系:软件i依赖软件Di.现在请你设计出

HDU5739 Fantasia 树形dp + 点双缩点

这个题当时打多校的时候有思路,但是代码能力差,没有写出来 事后看zimpha巨巨的题解,看了觉得基本差不多 核心思路:就是找出割点,然后变成森林,然后树形dp就可以搞了 关键就在重新构图上,缩完点以后,一个割点至少在两个点双里面,这个时候 把割点拿出来,分别和点双连边,也就是说,缩完的点双是不包含割点的,这个可以人为搞一下 (像有的点双里面只包含一个桥边,如果把割点拿出来,点双里面没有点了,这个时候把点双的权值积设为1就好) 然后说是树形dp,其实就是逆元搞一搞,这个很简单,树形dp只处理割点的

[树形dp][Tarjan][单调队列] Bzoj 1023 cactus仙人掌图

Description 如果某个无向连通图的任意一条边至多只出现在一条简单回路(simple cycle)里,我们就称这张图为仙人掌 图(cactus).所谓简单回路就是指在图上不重复经过任何一个顶点的回路. 举例来说,上面的第一个例子是一张仙人图,而第二个不是——注意到它有三条简单回路:(4,3,2,1,6 ,5,4).(7,8,9,10,2,3,7)以及(4,3,7,8,9,10,2,1,6,5,4),而(2,3)同时出现在前两 个的简单回路里.另外,第三张图也不是仙人图,因为它并不是连通图

代码风格与树形DP

Streaming很惨,不过因为比赛之间没有提交过就没掉(或掉了)rating.第二题是一个树形DP,但是我都在想第一题了,简直作死. 看着神犇的代码我也是醉了...各种宏,真是好好写会死系列. 看到他们Tree DP都用的DFS,突然感觉我这个蒟蒻的生活中充满了无力... 我一般都喜欢用BFS进行Tree DP.这样坏处很多,难调试,容易爆空间等.好处也有,写起来快,代码短,跑得飞快,判重简单.不过这样做是有条件的,而今天的Streaming这题就是一道可以的题. 然后讲讲今天这道LCASta

hdoj 2242 考研路茫茫——空调教室 【无向图求边双联通 缩点 + 树形dp】

考研路茫茫--空调教室 Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 2447    Accepted Submission(s): 721 Problem Description 众所周知,HDU的考研教室是没有空调的,于是就苦了不少不去图书馆的考研仔们.Lele也是其中一个.而某教室旁边又摆着两个未装上的空调,更是引起人们无限YY

HDU 2242 连通分量缩点+树形dp

题目大意是: 所有点在一个连通图上,希望去掉一条边得到两个连通图,且两个图上所有点的权值的差最小,如果没有割边,则输出impossible 这道题需要先利用tarjan算法将在同一连通分量中的点缩成一个点后,重新构建一幅图,然后利用新建的图进行树形dp解决问题 这道题目需要注意的是可能存在重边,那么子节点回到父节点有两条边及以上的话,就需要对子节点经过父节点的边进行low值更新 tarjan算法学习的网站个人感觉还不错https://www.byvoid.com/blog/scc-tarjan/