HDU1224-Free DIY Tour(SPFA求最长路+记录路径)

版权声明:转载请告知博主并要注明出处嗷~ https://blog.csdn.net/Akatsuki__Itachi/article/details/81746795

题目链接

把两条边的终点的分数作为此边的权值然后跑SPFA求最长路,用前驱数组pre记录下路径。

天快黑了,吃饭回宿舍,继续刷一刷题。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
#include<queue>
#define memset(a,v)  memset(a,v,sizeof(a))
using namespace std;
const int MAXL(100);
const int INF(0x3f3f3f3f);
const int mod(1e9+7);
struct node
{
    int to,next,w;
}edge[MAXL+50];
int head[MAXL*MAXL+50];
int vis[MAXL+50];
int value[MAXL+50];
int dis[MAXL+50];
int pre[MAXL+50];
int cnt=0,n,index=0;
void init()
{
    memset(head,-1);
    memset(vis,0);
    memset(dis,-INF);
    memset(pre,0);
    cnt=0;
}
void addedge(int x,int y,int w)
{
    edge[cnt].to=y;
    edge[cnt].next=head[x];
    edge[cnt].w=w;
    head[x]=cnt++;
}
int SPFA(int u)
{
    queue<int>q;
    q.push(u);
    vis[u]=1;
    dis[u]=0;
    for(int i=u;~i;i=edge[i].next)
        pre[edge[i].to]=u;
    while(!q.empty())
    {
        int v=q.front();
        q.pop(),vis[v]=0;
        for(int i=head[v];~i;i=edge[i].next)
        {
            int to=edge[i].to,w=edge[i].w;
            if(dis[to]<dis[v]+w)
            {
                dis[to]=dis[v]+w;
                pre[to]=v;
                if(!vis[to])
                    vis[to]=1,q.push(to);
            }
        }
    }
    return dis[n+1];
}
int main()
{
    int T,CASE=1;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",value+i);
        int m;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            addedge(x,y,value[y==n+1?1:y]);
        }
        int ans=SPFA(1);
        cout<<"CASE "<<CASE++<<"#"<<endl;
        stack<int>sk;
        int now=n+1;
        while(1)
        {
            sk.push(now);
            if(now==1) break;
            now=pre[now];
        }
        cout<<"points : "<<ans<<endl;
        cout<<"circuit : ";
        cout<<sk.top();
        sk.pop();
        while(!sk.empty())
        {
            cout<<"->"<<(sk.top()==(n+1)? 1:sk.top());
            sk.pop();
        }
        cout<<endl;
        if(T>0)
            cout<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/Akatsuki__Itachi/article/details/81746795