WA 很多次,原因出在inf设定太小了,以后果断用<climits>头文件中的LLONG_MAX,ULLONG_MAX,INT_MAX之类的表示无穷大吧,不然遇上这类坑题绝对给跪。
代码:
#include <iostream>
#include <cstdio>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
#define esp 1e-6
#define pb push_back
#define in freopen("in.txt", "r", stdin);
#define out freopen("out.txt", "w", stdout);
#define print(a) printf("%d\n",(a));
#define bug puts("********))))))");
#define Rep(i, c) for(__typeof(c.end()) i = c.begin(); i != c.end(); i++)
#define inf LLONG_MAX
#define INF 0x0f0f0f0f0f0f
#define N 100010
#define M 2000100
using namespace std;
typedef __int64 LL;
typedef vector<int> VI;
typedef vector<int>:: iterator IT;
struct EDGE{
int i, c, d;
EDGE *nxt, *ani;
} *Edge[N], *Path[N], E[M];
LL dis[N], cnt;
bool inq[N];
int n, m, sink, src;
VI ans;
void add(int i, int j, int c, int d, EDGE &e1, EDGE &e2)
{
e1.i = j, e1.c = c, e1.d = d, e1.nxt = Edge[i], e1.ani = &e2, Edge[i] = &e1;
e2.i = i, e2.c = 0, e2.d = -d, e2.nxt = Edge[j], e2.ani = &e1, Edge[j] = &e2;
}
void init(void)
{
cnt = 0;
memset(Edge, 0, sizeof(Edge));
}
bool SPFA(int s, int end)
{
for(int i = 0; i < N; i++)
dis[i] = inf;
queue<int> q;
int Max = dis[0];
dis[s] = 0;
inq[s] = true;
q.push(s);
while(!q.empty())
{
int u = q.front();
int v;
q.pop();
inq[u] = 0;
for(EDGE *p = Edge[u]; p; p = p->nxt)
{
if(p->c && dis[v = p->i] > dis[u] + p->d)
if(Path[v] = p, dis[v] = dis[u] + p->d, !inq[v])
inq[v] = 1, q.push(v);
}
}
return dis[end] < inf;
}
bool min_cost_flow(int s, int end)
{
if(SPFA(s, end))
{
for(EDGE *p = Path[end]; p; p = Path[p->ani->i])
{
if(p->i != sink)
ans.pb(p->i);
}
return 1;
}
else return 0;
}
int main(void)
{
scanf("%d%d", &n, &m);
src = 0, sink = n + 1;
for(int i = 1; i <= m; i++)
{
int u, v, w;
scanf("%d%d%d", &u, &v, &w);
add(u, v, 1, w, E[cnt], E[cnt + 1]);
cnt += 2;
add(v, u, 1, w, E[cnt], E[cnt + 1]);
cnt += 2;
}
add(src, 1, 1, 0, E[cnt], E[cnt + 1]);
cnt += 2;
add(n, sink, 1, 0, E[cnt], E[cnt + 1]);
cnt += 2;
int ret = min_cost_flow(src, sink);
if(ret)
{
for(int i = ans.size() - 1; i >= 0; i--)
printf("%d%c", ans[i], (i == 0) ? '\n' : ' ');
}
else puts("-1");
return 0;
}