luoguP2756 飞行员配对方案问题(最大流)

题目传送门

很典型的二分图匹配问题
外籍飞行员作为X部,英国飞行员作为Y部
源点向X部连边,容量为1
Y部向汇点连边,容量为1
X部与Y部之间连边,容量为1

狂WA不止原来是因为方案输出
我这里用了一个数组记录(外籍飞行员,英国飞行员)边的编号
数组默认值为0
dinic之后判断边的容量是否为0
注意需要判断数组中记录的编号是否有效(是否为默认值)

Tip

注意bfs和dfs的时候,需要判断当前边是否还有容量

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
using namespace std;

const int INF=0x33333333;
const int N=105;
int mp[N][N];
int st[N],tot=-1;
struct node{
    int x,y,v,nxt;
};
node way[N*N];
int n,m,s,t;

void addway(int u,int w,int z) {
    tot++;
    way[tot].x=u; 
    way[tot].y=w;
    way[tot].v=z;
    way[tot].nxt=st[u];
    st[u]=tot;
}

queue<int> q;
int cur[N];
int deep[N];
int bfs() {
    memset(deep,-1,sizeof(deep));
    for (int i=0;i<=n+1;++i) cur[i]=st[i];
    q.push(s);
    deep[s]=1;
    while (!q.empty()) {
        int r=q.front();
        q.pop();
        for (int i=st[r];i!=-1;i=way[i].nxt) {
            int y=way[i].y;
            if (way[i].v && deep[y]==-1) {   //way[i].v!=0
                q.push(y);
                deep[y]=deep[r]+1;
            }
        }
    }
    return deep[t]!=-1;
}

int dfs(int now,int t,int limit) {
    if (now==t || !limit) return limit;
    int f,flow=0;
    for (int i=cur[now];i!=-1;i=way[i].nxt) {
        cur[now]=i;
        if (way[i].v && deep[way[i].y]==deep[now]+1 && (f=dfs(way[i].y,t,min(limit,way[i].v)))) {
            way[i].v-=f;
            way[i^1].v+=f;
            limit-=f;
            flow+=f;
            if (!limit) break;
        }
    }
    return flow;
}

int dinic() {
    int ans=0;
    while (bfs()) {
        ans+=dfs(s,t,INF);
    }
    return ans;
}

int main()
{
    memset(st,-1,sizeof(st));
    scanf("%d%d",&m,&n);
    s=0;
    t=n+1;
    for (int i=1;i<=n;i++) {
        if (i<=m) {
            addway(s,i,1);
            addway(i,s,0);
        }
        else {
            addway(i,t,1);
            addway(t,i,0);
        }
    }
    int x,y;
    scanf("%d%d",&x,&y);
    while (x!=-1 && y!=-1) {
        addway(x,y,1);
        mp[x][y]=tot;
        addway(y,x,0);
        scanf("%d%d",&x,&y);
    }
    printf("%d\n",dinic());
    for (int i=1;i<=m;++i) {
        for (int j=m+1;j<=n;++j) {
            if (mp[i][j] && way[mp[i][j]].v == 0) {
                printf("%d %d\n",i,j);
            }    
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
网络算法可以用来解决最大流最小割问题,而飞行员配对问题可以转化成最大流问题。下面是一个基于 Python 的网络算法求解飞行员配对方案问题的示例代码: ```python from collections import defaultdict class Graph: def __init__(self): self.graph = defaultdict(dict) def add_edge(self, u, v, w): self.graph[u][v] = w def bfs(self, s, t, parent): visited = set() queue = [s] visited.add(s) while queue: u = queue.pop(0) for v in self.graph[u]: if v not in visited and self.graph[u][v] > 0: queue.append(v) visited.add(v) parent[v] = u return True if t in visited else False def max_flow(self, source, sink): parent = {} max_flow = 0 while self.bfs(source, sink, parent): path_flow = float('inf') s = sink while s != source: path_flow = min(path_flow, self.graph[parent[s]][s]) s = parent[s] max_flow += path_flow v = sink while v != source: u = parent[v] self.graph[u][v] -= path_flow self.graph[v][u] += path_flow v = parent[v] return max_flow def pilot_pairing(pilots, planes): num_pilots = len(pilots) num_planes = len(planes) graph = Graph() source = 's' sink = 't' for i in range(num_pilots): graph.add_edge(source, 'P{}'.format(i), 1) for j in range(num_planes): if pilots[i][0] <= planes[j][0] and pilots[i][1] >= planes[j][1]: graph.add_edge('P{}'.format(i), 'A{}'.format(j), 1) graph.add_edge('A{}'.format(i), sink, 1) return graph.max_flow(source, sink) # 示例用法 pilots = [(1, 5), (2, 4), (3, 6), (2, 5)] planes = [(1, 4), (2, 5), (3, 6)] print(pilot_pairing(pilots, planes)) # 输出为 3 ``` 在上面的示例代码中,我们定义了一个 `Graph` 类来表示图,其中 `add_edge` 方法用于添加边,`bfs` 方法用于执行广度优先搜索,`max_flow` 方法用于计算最大流。 在 `pilot_pairing` 函数中,我们首先创建了一个 `Graph` 对象,然后为源点和汇点分别添加一条边,并遍历所有的飞行员和飞机,如果某个飞行员可以驾驶某个飞机,则在他们之间添加一条边。最后调用 `max_flow` 方法计算最大流。 在 `max_flow` 方法中,我们首先执行广度优先搜索来寻找一条增广路径,然后计算路径上的最小剩余容量,更新路径上的边的量,并更新最大流的值。重复执行这个过程,直到没有增广路径为止。最后返回最大流的值。 在上面的示例中,我们输入了 4 个飞行员和 3 个飞机,输出得到了最大流为 3,说明有 3 对飞行员和飞机可以配对成功。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值