洛谷 P2656 采蘑菇 树形DP+缩点+坑点

题目链接 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
#include
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<

你可能感兴趣的:(洛谷 P2656 采蘑菇 树形DP+缩点+坑点)