E. President and Roads (CF Round #Pi (Div. 2)) 优先队列的Dij+Tarjan找桥

23 篇文章 0 订阅
6 篇文章 0 订阅

E. President and Roads
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

Berland has n cities, the capital is located in city s, and the historic home town of the President is in city t (s ≠ t). The cities are connected by one-way roads, the travel time for each of the road is a positive integer.

Once a year the President visited his historic home town t, for which his motorcade passes along some path from s to t (he always returns on a personal plane). Since the president is a very busy man, he always chooses the path from s to t, along which he will travel the fastest.

The ministry of Roads and Railways wants to learn for each of the road: whether the President will definitely pass through it during his travels, and if not, whether it is possible to repair it so that it would definitely be included in the shortest path from the capital to the historic home town of the President. Obviously, the road can not be repaired so that the travel time on it was less than one. The ministry of Berland, like any other, is interested in maintaining the budget, so it wants to know the minimum cost of repairing the road. Also, it is very fond of accuracy, so it repairs the roads so that the travel time on them is always a positive integer.

Input

The first lines contain four integers nms and t (2 ≤ n ≤ 105; 1 ≤ m ≤ 105; 1 ≤ s, t ≤ n) — the number of cities and roads in Berland, the numbers of the capital and of the Presidents' home town (s ≠ t).

Next m lines contain the roads. Each road is given as a group of three integers ai, bi, li (1 ≤ ai, bi ≤ nai ≠ bi; 1 ≤ li ≤ 106) — the cities that are connected by the i-th road and the time needed to ride along it. The road is directed from city ai to city bi.

The cities are numbered from 1 to n. Each pair of cities can have multiple roads between them. It is guaranteed that there is a path froms to t along the roads.

Output

Print m lines. The i-th line should contain information about the i-th road (the roads are numbered in the order of appearance in the input).

If the president will definitely ride along it during his travels, the line must contain a single word "YES" (without the quotes).

Otherwise, if the i-th road can be repaired so that the travel time on it remains positive and then president will definitely ride along it, print space-separated word "CAN" (without the quotes), and the minimum cost of repairing.

If we can't make the road be such that president will definitely ride along it, print "NO" (without the quotes).

Sample test(s)
input
6 7 1 6
1 2 2
1 3 10
2 3 7
2 4 8
3 5 3
4 5 2
5 6 1
output
YES
CAN 2
CAN 1
CAN 1
CAN 1
CAN 1
YES
input
3 3 1 3
1 2 10
2 3 10
1 3 100
output
YES
YES
CAN 81
input
2 2 1 2
1 2 1
1 2 2
output
YES
NO
Note

The cost of repairing the road is the difference between the time needed to ride along it before and after the repairing.

In the first sample president initially may choose one of the two following ways for a ride: 1 → 2 → 4 → 5 → 6 or1 → 2 → 3 → 5 → 6.


题意:n个点m条有向边,总统从起点s到终点t走最短路,问哪些边是一定会走的(输出YES),哪些边通过减少距离可以使它一定被经过(输出减少的距离),否则输出NO。

思路:正反两遍求最短路,然后根据dist把那些最短路上的边重新建图,注意是无向图,然后用Tarjan求割边,最后每条边判断输出。居然卡SPFA,改成Dij就过了,以后就用Dij吧,恩。。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <string>
#include <map>
#include <stack>
#include <vector>
#include <set>
#include <queue>
#pragma comment (linker,"/STACK:102400000,102400000")
#define pi acos(-1.0)
#define eps 1e-6
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define FRE(i,a,b)  for(i = a; i <= b; i++)
#define FREE(i,a,b) for(i = a; i >= b; i--)
#define FRL(i,a,b)  for(i = a; i < b; i++)
#define FRLL(i,a,b) for(i = a; i > b; i--)
#define mem(t, v)   memset ((t) , v, sizeof(t))
#define sf(n)       scanf("%d", &n)
#define sff(a,b)    scanf("%d %d", &a, &b)
#define sfff(a,b,c) scanf("%d %d %d", &a, &b, &c)
#define pf          printf
#define DBG         pf("Hi\n")
typedef __int64 ll;
using namespace std;

#define INF 0x3f3f3f3f
#define mod 1000000009
const int maxn = 1005;
const int MAXN = 100011;
const int MAXM = 250010;
const int N = 1005;

int n,m,s,t;

struct Edge
{
    int u,v,next,id;
    ll len;
}e1[MAXM],e2[MAXM];

int head1[MAXN],head2[MAXN];

typedef pair<ll,int>P;
ll dist1[MAXN],dist2[MAXN];
int num,tot,start,End;
bool cut[MAXN];

int Low[MAXN],DFN[MAXN],Stack[MAXN];
bool Instack[MAXN];
int Index,top;

void init()
{
    num=0;  tot=0;
    mem(head1,-1);
    memset(head2,-1,sizeof(head2));
}

void addedge(int u,int v,ll w,int id)
{
    e1[num].u=u;
    e1[num].v=v;
    e1[num].len=w;
    e1[num].id=id;
    e1[num].next=head1[u];
    head1[u]=num++;
}

void addedge2(int u,int v,ll w,int id)
{
    e2[tot].u=u;
    e2[tot].v=v;
    e2[tot].len=w;
    e2[tot].id=id;
    e2[tot].next=head2[u];
    head2[u]=tot++;
}

void Dijkstra(int s,int t,int head[],ll dist[],Edge edge[])
{
    priority_queue<P,vector<P>,greater<P> >Q;
    dist[s]=0;
    Q.push(make_pair(0,s));
    while (!Q.empty())
    {
        P p=Q.top(); Q.pop();
        int u=p.second;
        if (dist[u]<p.first) continue;
        for (int i=head[u];~i;i=edge[i].next)
        {
            int v=edge[i].v;
            if (dist[v]>dist[u]+edge[i].len)
            {
                dist[v]=dist[u]+edge[i].len;
                Q.push(make_pair(dist[v],v));
            }
        }
    }
}

void Tarjan(int u,int pre)
{
    int v;
    Low[u]=DFN[u]=++Index;
    Stack[top++]=u;
    Instack[u]=true;
    int pre_cnt=0;
    for (int i=head2[u];~i;i=e2[i].next)
    {
        v=e2[i].v;
        if (v==pre&&pre_cnt==0) { pre_cnt++;continue; }
        if (!DFN[v])
        {
            Tarjan(v,u);
            if (Low[u]>Low[v]) Low[u]=Low[v];
            if (Low[v]>DFN[u])
                cut[e2[i].id]=true;
        }
        else if (Low[u]>DFN[v])
            Low[u]=DFN[v];
    }
    Instack[u]=false;
    top--;
}

void solve()
{
    memset(DFN,0,sizeof(DFN));
    memset(Instack,false,sizeof(Instack));
    Index=top=0;
    for (int i=1;i<=n;i++)
        if (!DFN[i])
        Tarjan(i,i);
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("C:/Users/lyf/Desktop/IN.txt","r",stdin);
#endif
    int i,j,u,v;
    ll w;
    while (~scanf("%d%d%d%d",&n,&m,&s,&t))
    {
        init();
        for (i=0;i<m;i++)
        {
            scanf("%d%d%I64d",&u,&v,&w);
            addedge(u,v,w,i);
            addedge2(v,u,w,i);
        }
        mem(dist1,INF);
        mem(dist2,INF);
        Dijkstra(s,t,head1,dist1,e1);
        Dijkstra(t,s,head2,dist2,e2);
        tot=0;
        memset(head2,-1,sizeof(head2));
        for (i=0;i<m;i++)
        {
            cut[i]=false;
            u=e1[i].u;
            v=e1[i].v;
            if (dist1[u]+dist2[v]+e1[i].len==dist1[t])
            {
                addedge2(u,v,e1[i].len,i);
                addedge2(v,u,e1[i].len,i);
            }
        }
        solve();
        for (i=0;i<m;i++)
        {
            u=e1[i].u;
            v=e1[i].v;
            if (cut[i]) printf("YES\n");
            else if (dist1[u]+dist2[v]+e1[i].len==dist1[t])
            {
                if (e1[i].len>1)
                    printf("CAN 1\n");
                else
                    printf("NO\n");
            }
            else
            {
                ll dd=dist1[t]-dist1[u]-dist2[v]-1;
                if (dd<=0)
                    printf("NO\n");
                else
                    printf("CAN %I64d\n",e1[i].len-dd);
            }
        }
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值