HDU 1224 Free DIY Tour SPFA

题目传送门：Problem - 1224

样例输入：

2

4
0 1000 100 10
4
1 2
2 3
3 4
4 5

5

0 100 200 300 400

5

1 2

1 3

2 3

3 5

5 6

样例输出：

CASE 1#

points : 1110

circuit : 1->2->3->4->1

CASE 2#

points : 700

circuit : 1->2->3->5->1

题目大意：

        给你一张有向图，每个城市一个权重，计算从走到n+1城市所经过城市的权重合最大值，同时输出路径。

解题思路：

        利用SPFA跑一遍最长路，笔者一开始用的dijkstra跑的，但是wa了(笔者太菜了，至今不知道是算法本身的局限性还是我写丑了)，然后改用SPFA跑就过了，板子题，将SPFA板子copy过来，改一下条件，最短路瞬间变成最长路。同时利用vector存储跑到第i个城市权重最大的路线，有个简单的状态转移，看代码。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int N=110;
const int M=1e4+10;
int n,hh[N],m,id,maxd,t,tt;
int h[N],e[M],ne[M],w[M],idx;
bool st[N],flag[N];
int dis[N];
vector<int> v[N];
void add(int a,int b)
{
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=hh[b];
    h[a]=idx++;
}
void init()
{
    idx=0;
    id=1;
    maxd=0;
    for (int i=1; i<=n; i++) {
        v[i].clear();
    }
    memset(h, -1, sizeof(h));
    memset(st, 0, sizeof(st));
    memset(dis, 0, sizeof(dis));
    memset(flag, 0, sizeof(flag));
}
void spfa()
{
    queue<int>q;
    q.push(1);
    st[1]=true;
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        st[t]=false;
        for (int i=h[t]; i!=-1; i=ne[i]) {
            int j=e[i];
            if(dis[j]<dis[t]+w[i])
            {
                dis[j]=dis[t]+w[i];
                v[j]=v[t];
                v[j].push_back(j);
                if(!st[j])
                {
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }
}
void get_date()
{
    scanf("%d",&n);
    for (int i=1; i<=n; i++) {
        scanf("%d",&hh[i]);
        //v[i].push_back(1);
    }
    v[1].push_back(1);
    scanf("%d",&m);
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(min(a,b), max(a, b));
        if(max(a, b)==n+1) flag[min(a, b)]=true;
    }
}
void get_ans()
{
    for (int i=1; i<=n; i++) {
        if(dis[i]>maxd&&flag[i])
        {
            maxd=dis[i];
            id=i;
        }
    }
}
void solve()
{
    printf("CASE %d#\npoints : %d\ncircuit : ",tt,maxd);
    for (int i=0; i<v[id].size(); i++) {
        printf("%d->",v[id][i]);
    }
    printf("1\n");
    if(t!=0) printf("\n");
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        tt++;
        init();
        get_date();
        spfa();
        get_ans();
        solve();
    }
    return 0;
}