洛谷 P2656 采蘑菇 tarjan 缩点 spfa

题目链接：

https://www.luogu.org/problem/P2656

思路：

1：tarjan缩点后，如果同一种颜色的节点个数大于1，即一个强连通分量所包含的节点个数大于1，那么就去榨干这一个强连通分量所包含的每一条边，记作，缩点之后，就是把每一个强连通分量看作一个节点的图中这个节点的点权

 for(int i=1;i<=cnt;i++)
    {
        if(cc[i].size()>1)
        {
             for(int j=1;j<=n;j++)
             {
                 if(color[j]==i)
                 {
                     for(int k=0;k<es[j].size();k++)
                     {
                         if(color[(int)es[j][k][0]]==i)
                         {
                             colorcnt[i]+=es[j][k][1];
                             double w=es[j][k][1];
                             w=floor(w*es[j][k][2]);
                             while(w!=0)
                             {
                                 colorcnt[i]+=w;
                                 w=floor(w*es[j][k][2]);
                             }
                         }
                     }
                 }
             }
        }
    }

2：跑spfa求最长路径，这也时自己一开始30分，纠结好几天之后才发现有以下注意点，对应三种解法

    解法一：

    1：spfa函数内的判断条件是

 

int v=mapp[now][i].first;
            if(d[v]>d[now]+mapp[now][i].second-colorcnt[v])
            {
               d[v]=d[now]+mapp[now][i].second-colorcnt[v];
                if(ing[v])
                    continue;
                q.push(v);
                ing[v]=1;
            }

d[now]就代表到这一点的最短路径（可以采到的最多蘑菇数），那么它和它的下一个节点d[v]的差别就是节点now到节点v路上可以采的蘑菇加上在节点v可以采的蘑菇

2：在缩点之后建图的过程中要区别与没有用到spfa算法的题目，不应该去判断如果两个强连通分量之间有边就不再建边了，应该把两个强连通分量所有的边都建起来，因为跑spfa时会走最短路径（这里要特别注意一下，无论是边权还是点权存的都是负值，跑完最短路，取相反数就是最长路

  for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
//            if(color[i]!=color[(int)es[i][j][0]]&&ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
//            {
//                ee[color[i]].insert(color[(int)es[i][j][0]]);
//                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]));
//            }
            if(color[i]!=color[(int)es[i][j][0]])
            {
                ee[color[i]].insert(color[(int)es[i][j][0]]);
                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]));
            }
        }
    }

#include <bits/stdc++.h>

using namespace std;
const int maxn=8e4+1;
vector<double>e;
vector<vector<double> >es[maxn];
set<int>ee[maxn];
int dfn[maxn],low[maxn],ins[maxn],color[maxn],timing,cnt,n,m,f;
double colorcnt[maxn];
stack<int>s;
vector<int>cc[maxn];
vector<pair<int,double> >mapp[maxn];
double d[maxn],ing[maxn];

void tarjan(int x)
{
    low[x]=dfn[x]=++timing;
    s.push(x);
    ins[x]=1;
    for(int i=0;i<es[x].size();i++)
    {
        int v=es[x][i][0];
        if(!dfn[v])
        {
            tarjan(v);
            low[x]=min(low[x],low[v]);
        }
        else if(ins[v])
        {
            low[x]=min(low[x],dfn[v]);
        }
    }
    if(dfn[x]==low[x])
    {
        cnt++;
        while(s.top()!=x)
        {
            color[s.top()]=cnt;
            ins[s.top()]=0;
            cc[cnt].push_back(s.top());
            s.pop();
        }
        color[s.top()]=cnt;
        ins[s.top()]=0;
        cc[cnt].push_back(s.top());
        s.pop();
    }
}

void init()
{
    for(int i=1;i<=cnt;i++)
    {
        ing[i]=0;
        d[i]=1e9;
    }
}

int spfa(int x)
{
    queue<int>q;
    q.push(x);
    d[x]=-colorcnt[x];
    ing[x]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        ing[now]=0;
        for(int i=0;i<mapp[now].size();i++)
        {
            int v=mapp[now][i].first;
            if(d[v]>d[now]+mapp[now][i].second-colorcnt[v])
            {
               d[v]=d[now]+mapp[now][i].second-colorcnt[v];
                if(ing[v])
                    continue;
                q.push(v);
                ing[v]=1;
            }
        }
    }
    double mi=1e9;
    for(int i=1;i<=cnt;i++)
    {
        mi=min(mi,d[i]);
    }
    return -mi;
}

int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        double a,b,c,d;
        e.clear();
        cin>>a>>b>>c>>d;
        e.push_back(b);
        e.push_back(c);
        e.push_back(d);
        es[(int)a].push_back(e);
    }
    cin>>f;
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }
    for(int i=1;i<=cnt;i++)
    {
        if(cc[i].size()>1)
        {
             for(int j=1;j<=n;j++)
             {
                 if(color[j]==i)
                 {
                     for(int k=0;k<es[j].size();k++)
                     {
                         if(color[(int)es[j][k][0]]==i)
                         {
                             colorcnt[i]+=es[j][k][1];
                             double w=es[j][k][1];
                             w=floor(w*es[j][k][2]);
                             while(w!=0)
                             {
                                 colorcnt[i]+=w;
                                 w=floor(w*es[j][k][2]);
                             }
                         }
                     }
                 }
             }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
//            if(color[i]!=color[(int)es[i][j][0]]&&ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
//            {
//                ee[color[i]].insert(color[(int)es[i][j][0]]);
//                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]));
//            }
            if(color[i]!=color[(int)es[i][j][0]])
            {
                ee[color[i]].insert(color[(int)es[i][j][0]]);
                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]));
            }
        }
    }
    init();
    cout<<spfa(color[f])<<endl;
    return 0;
}

解法二：

1：建图时，就直接把边的终点的那个节点的点权值，加入到这条边的权值中去

for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
//            if(color[i]!=color[(int)es[i][j][0]]&&ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
//            {
//                ee[color[i]].insert(color[(int)es[i][j][0]]);
//                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
//            }
            if(color[i]!=color[(int)es[i][j][0]])
            {
                ee[color[i]].insert(color[(int)es[i][j][0]]);
                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
            }
        }
    }

2：然后跑spfa时就是一般的判断条件了

            int v=mapp[now][i].first;
            if(d[v]>d[now]+mapp[now][i].second)
            {
               //d[v]>d[now]+mapp[now][i].second-colorcnt[now]
               // d[v]=d[now]+mapp[now][i].second-colorcnt[now]-colorcnt[v];
                d[v]=d[now]+mapp[now][i].second;
                if(ing[v])
                    continue;
                q.push(v);
                ing[v]=1;
            }

#include <bits/stdc++.h>

using namespace std;
const int maxn=8e4+1;
vector<double>e;
vector<vector<double> >es[maxn];
set<int>ee[maxn];
int dfn[maxn],low[maxn],ins[maxn],color[maxn],timing,cnt,n,m,f;
double colorcnt[maxn];
stack<int>s;
vector<int>cc[maxn];
vector<pair<int,double> >mapp[maxn];
double d[maxn],ing[maxn];

void tarjan(int x)
{
    low[x]=dfn[x]=++timing;
    s.push(x);
    ins[x]=1;
    for(int i=0;i<es[x].size();i++)
    {
        int v=es[x][i][0];
        if(!dfn[v])
        {
            tarjan(v);
            low[x]=min(low[x],low[v]);
        }
        else if(ins[v])
        {
            low[x]=min(low[x],dfn[v]);
        }
    }
    if(dfn[x]==low[x])
    {
        cnt++;
        while(s.top()!=x)
        {
            color[s.top()]=cnt;
            ins[s.top()]=0;
            cc[cnt].push_back(s.top());
            s.pop();
        }
        color[s.top()]=cnt;
        ins[s.top()]=0;
        cc[cnt].push_back(s.top());
        s.pop();
    }
}

void init()
{
    for(int i=1;i<=cnt;i++)
    {
        ing[i]=0;
        d[i]=1e9;
    }
}

int spfa(int x)
{
    queue<int>q;
    q.push(x);
    d[x]=-colorcnt[x];
    ing[x]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        ing[now]=0;
        for(int i=0;i<mapp[now].size();i++)
        {
            int v=mapp[now][i].first;
            if(d[v]>d[now]+mapp[now][i].second)
            {
               //d[v]>d[now]+mapp[now][i].second-colorcnt[now]
               // d[v]=d[now]+mapp[now][i].second-colorcnt[now]-colorcnt[v];
                d[v]=d[now]+mapp[now][i].second;
                if(ing[v])
                    continue;
                q.push(v);
                ing[v]=1;
            }
        }
    }
    double mi=1e9;
    for(int i=1;i<=cnt;i++)
    {
        mi=min(mi,d[i]);
    }
    return -mi;
}

int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        double a,b,c,d;
        e.clear();
        cin>>a>>b>>c>>d;
        e.push_back(b);
        e.push_back(c);
        e.push_back(d);
        es[(int)a].push_back(e);
    }
    cin>>f;
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }
    for(int i=1;i<=cnt;i++)
    {
        if(cc[i].size()>1)
        {
             for(int j=1;j<=n;j++)
             {
                 if(color[j]==i)
                 {
                     for(int k=0;k<es[j].size();k++)
                     {
                         if(color[(int)es[j][k][0]]==i)
                         {
                             colorcnt[i]+=es[j][k][1];
                             double w=es[j][k][1];
                             w=floor(w*es[j][k][2]);
                             while(w!=0)
                             {
                                 colorcnt[i]+=w;
                                 w=floor(w*es[j][k][2]);
                             }
                         }
                     }
                 }
             }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
//            if(color[i]!=color[(int)es[i][j][0]]&&ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
//            {
//                ee[color[i]].insert(color[(int)es[i][j][0]]);
//                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
//            }
            if(color[i]!=color[(int)es[i][j][0]])
            {
                ee[color[i]].insert(color[(int)es[i][j][0]]);
                mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
            }
        }
    }
    init();
    cout<<spfa(color[f])<<endl;
    return 0;
}

解法三：

1：缩点之后建图的过程中就直接判断找出两个节点之间边的权值的最大值

for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
            if(color[i]!=color[(int)es[i][j][0]])
            {
                if(ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
                {
                    ee[color[i]].insert(color[(int)es[i][j][0]]);
                    mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
                }
                else
                {
                    for(int k=0;k<mapp[color[i]].size();k++)
                    {
                        int v=mapp[color[i]][k].first;
                        if(v==color[(int)es[i][j][0]])
                        {
                            if(mapp[color[i]][k].second>-es[i][j][1]-colorcnt[(int)es[i][j][0]])
                            {
                                mapp[color[i]][k].second=-es[i][j][1]-colorcnt[(int)es[i][j][0]];
                            }
                        }
                    }
                }
            }
        }
    }

#include <bits/stdc++.h>

using namespace std;
const int maxn=8e4+1;
vector<double>e;
vector<vector<double> >es[maxn];
set<int>ee[maxn];
int dfn[maxn],low[maxn],ins[maxn],color[maxn],timing,cnt,n,m,f;
double colorcnt[maxn];
stack<int>s;
vector<int>cc[maxn];
vector<pair<int,double> >mapp[maxn];
double d[maxn],ing[maxn];

void tarjan(int x)
{
    low[x]=dfn[x]=++timing;
    s.push(x);
    ins[x]=1;
    for(int i=0;i<es[x].size();i++)
    {
        int v=es[x][i][0];
        if(!dfn[v])
        {
            tarjan(v);
            low[x]=min(low[x],low[v]);
        }
        else if(ins[v])
        {
            low[x]=min(low[x],dfn[v]);
        }
    }
    if(dfn[x]==low[x])
    {
        cnt++;
        while(s.top()!=x)
        {
            color[s.top()]=cnt;
            ins[s.top()]=0;
            cc[cnt].push_back(s.top());
            s.pop();
        }
        color[s.top()]=cnt;
        ins[s.top()]=0;
        cc[cnt].push_back(s.top());
        s.pop();
    }
}

void init()
{
    for(int i=1;i<=cnt;i++)
    {
        ing[i]=0;
        d[i]=1e9;
    }
}

int spfa(int x)
{
    queue<int>q;
    q.push(x);
    d[x]=-colorcnt[x];
    ing[x]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        ing[now]=0;
        for(int i=0;i<mapp[now].size();i++)
        {
            int v=mapp[now][i].first;
            if(d[v]>d[now]+mapp[now][i].second)
            {
               d[v]=d[now]+mapp[now][i].second;
                if(ing[v])
                    continue;
                q.push(v);
                ing[v]=1;
            }
        }
    }
    double mi=1e9;
    for(int i=1;i<=cnt;i++)
    {
        mi=min(mi,d[i]);
    }
    return -mi;
}

int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        double a,b,c,d;
        e.clear();
        cin>>a>>b>>c>>d;
        e.push_back(b);
        e.push_back(c);
        e.push_back(d);
        es[(int)a].push_back(e);
    }
    cin>>f;
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])
            tarjan(i);
    }
    for(int i=1;i<=cnt;i++)
    {
        if(cc[i].size()>1)
        {
             for(int j=1;j<=n;j++)
             {
                 if(color[j]==i)
                 {
                     for(int k=0;k<es[j].size();k++)
                     {
                         if(color[(int)es[j][k][0]]==i)
                         {
                             colorcnt[i]+=es[j][k][1];
                             double w=es[j][k][1];
                             w=floor(w*es[j][k][2]);
                             while(w!=0)
                             {
                                 colorcnt[i]+=w;
                                 w=floor(w*es[j][k][2]);
                             }
                         }
                     }
                 }
             }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<es[i].size();j++)
        {
            if(color[i]!=color[(int)es[i][j][0]])
            {
                if(ee[color[i]].find(color[(int)es[i][j][0]])==ee[color[i]].end())
                {
                    ee[color[i]].insert(color[(int)es[i][j][0]]);
                    mapp[color[i]].push_back(make_pair(color[(int)es[i][j][0]],-es[i][j][1]-colorcnt[(int)es[i][j][0]]));
                }
                else
                {
                    for(int k=0;k<mapp[color[i]].size();k++)
                    {
                        int v=mapp[color[i]][k].first;
                        if(v==color[(int)es[i][j][0]])
                        {
                            if(mapp[color[i]][k].second>-es[i][j][1]-colorcnt[(int)es[i][j][0]])
                            {
                                mapp[color[i]][k].second=-es[i][j][1]-colorcnt[(int)es[i][j][0]];
                            }
                        }
                    }
                }
            }
        }
    }
    init();
    cout<<spfa(color[f])<<endl;
    return 0;
}