洛谷 P2656 采蘑菇 树形DP+缩点+坑点
- 2020 年 4 月 4 日
- 筆記
题目链接 https://www.luogu.com.cn/problem/P2656
分析
这其实是个一眼题（bushi
发现如果没有那个恢复系数，缩个点就完了，有恢复系数呢？你发现这个恢复系数其实在DAG中没有用，因为走不回去不管怎么恢复都没啥用，所以对于走不回去的子图没有什么用，于是就想到了缩点，把每个强连通缩成一个点就完了，因为我能恢复的话肯定走的越多越好，所以就把每个强连通都榨干就完了，统计答案就dp一下，正好刚学的树形dp，所以大概思路就有了。
我们先通过tarjan跑出强连通分量（有向图），然后缩点，最后dp，转移方程也挺简单的，dp[i]表示以i为跟的子树，初始化为W[i]
(dp[i]+=max（dp[v]）)我最开始想的版本
但是有一个问题，这么定义的话缩点前权值在边上，缩点后权值在点上，我起初的处理办法是将边权都压到边的终点，因为我只有走过这条边才能获得这个权值，乍一看是没啥问题，但是呢？的确如果从根开始dp不会有问题，但这道题是从某一不定的节点开始dp的，这样就会出问题。
比如这里，我t->s这条边的权值会被压到s点上，如果我从t开始dp，没问题，从s开始，明明没有走那条边，却加上了边权，WA。
解决这个问题很简单啊，就特判一下，同一个连通分量内的点把权值压在点上，另外的放在边上，dp方程改成
(dp[i]+=max（dp[v]+E.val）)
然后这个问题就解决了，这道题一开始Wa的主要原因还是点权边权的处理，当然也可能是没想太明白就开始打代码，导致出现问题，总结一下，以后要先想明白再写，想出来思路也不一定对
#include<iostream> #include<cstdio> using namespace std; const int N=8e4+10,M=2e5+10; struct Edge{ int fro,nxt,to,val; double hui; }e[M],E[M]; int Head[N],len; void Ins(int a,int b,int c,double d){ e[++len].fro=a;e[len].to=b;e[len].nxt=Head[a]; Head[a]=len;e[len].val=c;e[len].hui=d; } inline int read(){ char ch=getchar(); while(ch<'0'||ch>'9')ch=getchar(); int x=0; while(ch<='9'&&ch>='0'){ x=x*10+ch-'0'; ch=getchar(); } return x; } int dfn[N],low[N],belong[N],stk[N],top,scc_cnt,num; void tarjan(int u){ dfn[u]=low[u]=++num; stk[++top]=u; for(int i=Head[u];i;i=e[i].nxt){ int v=e[i].to; if(!dfn[v]){ tarjan(v); low[u]=min(low[u],low[v]); }else if(!belong[v])low[u]=min(low[u],dfn[v]); } if(low[u]==dfn[u]){ scc_cnt++; while(1){ int x=stk[top--]; belong[x]=scc_cnt; if(x==u)break; } } } int H[N],l,w[N]; void I(int a,int b,int c){ E[++l].to=b;E[l].nxt=H[a];H[a]=l;E[l].val=c; } int dp[N]; void dfs(int u){ if(dp[u])return; dp[u]=w[u]; int now=0; for(int x=H[u];x;x=E[x].nxt){ int v=E[x].to; dfs(v); now=max(now,dp[v]+E[x].val); } dp[u]+=now; } int main(){ int n,m; n=read();m=read(); for(int i=1;i<=m;i++){ int a,b,c;double d; a=read();b=read();c=read();cin>>d; Ins(a,b,c,d); } int s=read(); tarjan(s); for(int i=1;i<=m;i++){ int u=belong[e[i].fro],v=belong[e[i].to]; if(u!=v)I(u,v,e[i].val); if(u==v){ int now=e[i].val;double f=e[i].hui; while(now){ w[v]+=now; now=(int)now*f; } } } dfs(belong[s]); cout<<dp[belong[s]]; } 
