Gym - 101615G Security Badge(离散化+dfs)

题意:给出n个点,m条边的有向图,有条边都有一个区间限制,现在有k个人(编号 1 - k)需要从起点走到终点,每条边只有当这个人的编号在其范围内才能通过,求有多少人能走到终点。

分析:刚开始的做法是从起点到终点跑bfs,求每条边的区间交集,奈何k是1e9,MLE没得解释,猝~

之后在hzy学长的帮助下对这个题又有了新的理解,我们先把每个区间的端点存下来,然后给端点排序,然后对每一个端点跑dfs,看其能否从起点到达终点,如果这个点能到达,就说明一定有一个区间包括了这个端点,我们就用这个端点减去他的前一个端点(已经排好序),得到的这个区间一定是符合条件的区间,之后将这些区间大小相加即可,在离散化区间的时候,要注意区间的左端点要减1,因为如果出现的左界减右界的情况,这时候就会出问题,所以需要减1,例如下面的这组样例

3 3 5
1 2
1 2 1 2
1 3 4 5
3 2 4 5
可以自己验证一下

代码:

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <iomanip>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define mod 100000000
#define lowbit(x) (x&(-x))
#define mem(a,b) memset(a,b,sizeof(a))
#define FRER() freopen("in.txt","r",stdin);
#define FREW() freopen("out.txt","w",stdout);

using namespace std;

typedef pair<int,int> pii;
const int N = 1000 + 7 , M = 5000 + 7 , inf = 0x3f3f3f3f ;
int nxt[M],from[M],to[M],head[M],l[M],r[M],p[M*2],vis[N];
int n,m,k,nEdge,s,t,cnt,ans;
void AddEdge(int u,int v,int ll,int rr){
    from[nEdge] = u;
    to[nEdge]   = v;
    nxt[nEdge]  = head[u];
    l[nEdge]    = ll;
    r[nEdge]    = rr;
    head[u]     = nEdge ++;
}
void dfs(int u,int cur){
    vis[u] = 1;
    for(int e = head[u];~e;e=nxt[e]){
        int v = to[e] , ll = l[e] , rr = r[e];
        if(!vis[v]&&ll<=cur&&cur<=rr)dfs(v,cur);
    }
}

int main(){
    //FRER();
    mem(head,-1);
    nEdge = cnt = ans = 0;
    scanf("%d%d%d",&n,&m,&k);
    scanf("%d%d",&s,&t);
    for(int i=0;i<m;i++){
        int u,v,ll,rr;
        scanf("%d%d%d%d",&u,&v,&ll,&rr);
        AddEdge(u,v,ll,rr);
        p[++cnt] = ll-1 ;
        p[++cnt] = rr;
    }
    sort(p+1,p+cnt+1);
    for(int i=1;i<=cnt;i++){
        mem(vis,0);
        dfs(s,p[i]);
        if(vis[t]) ans+=p[i]-p[i-1];
    }
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Insist_77/article/details/82468034