56-实战演练
7.8.5 实战演练
//program 7-7
#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int INF=0x3fffffff;
const int N=100;
const int M=10000;
int top;
int h[N], pre[N], g[N];
bool flag[N];//标记选中的结点
struct Vertex
{
int first;
}V[N];
struct Edge
{
int v, next;
int cap, flow;
}E[M];
void init()
{
memset(V, -1, sizeof(V));
top = 0;
}
void add_edge(int u, int v, int c)
{
E[top].v = v;
E[top].cap = c;
E[top].flow = 0;
E[top].next = V[u].first;
V[u].first = top++;
}
void add(int u,int v, int c)
{
add_edge(u,v,c);
add_edge(v,u,0);
}
void set_h(int t,int n)
{
queue<int> Q;
memset(h, -1, sizeof(h));
memset(g, 0, sizeof(g));
h[t] = 0;
Q.push(t);
while(!Q.empty())
{
int v = Q.front(); Q.pop();
++g[h[v]];
for(int i = V[v].first; ~i; i = E[i].next)
{
int u = E[i].v;
if(h[u] == -1)
{
h[u] = h[v] + 1;
Q.push(u);
}
}
}
cout<<"初始化高度"<<endl;
cout<<"h[ ]=";
for(int i=1;i<=n;i++)
cout<<" "<<h[i];
cout<<endl;
}
int Isap(int s, int t,int n)
{
set_h(t,n);
int ans=0, u=s;
int d;
while(h[s]<n)
{
int i=V[u].first;
if(u==s)
d=INF;
for(; ~i; i=E[i].next)
{
int v=E[i].v;
if(E[i].cap>E[i].flow && h[u]==h[v]+1)
{
u=v;
pre[v]=i;
d=min(d, E[i].cap-E[i].flow);
if(u==t)
{
cout<<endl;
cout<<"增广路径:"<<t;
while(u!=s)
{
int j=pre[u];
E[j].flow+=d;
E[j^1].flow-=d;
u=E[j^1].v;
cout<<"--"<<u;
}
cout<<"增流:"<<d<<endl;
ans+=d;
d=INF;
}
break;
}
}
if(i==-1)
{
if(--g[h[u]]==0)
break;
int hmin=n-1;
for(int j=V[u].first; ~j; j=E[j].next)
if(E[j].cap>E[j].flow)
hmin=min(hmin, h[E[j].v]);
h[u]=hmin+1;
cout<<"重贴标签后高度"<<endl;
cout<<"h[ ]=";
for(int i=1;i<=n;i++)
cout<<" "<<h[i];
cout<<endl;
++g[h[u]];
if(u!=s)
u=E[pre[u]^1].v;
}
}
return ans;
}
void printg(int n)//输出网络邻接表
{
cout<<"----------网络邻接表如下:----------"<<endl;
for(int i=0;i<=n;i++)
{
cout<<"v"<<i<<" ["<<V[i].first;
for(int j=V[i].first;~j;j=E[j].next)
cout<<"]--["<<E[j].v<<" "<<E[j].cap<<" "<<E[j].flow<<" "<<E[j].next;
cout<<"]"<<endl;
}
}
void DFS(int s) //深度搜索最大获益方案
{
for(int i=V[s].first;~i;i=E[i].next)//读当前结点的邻接表
if(E[i].cap>E[i].flow)
{
int u=E[i].v;
if(!flag[u])
{
flag[u]=true;
DFS(u);
}
}
}
void print(int m,int n) //输出最佳方案
{
cout<<"----------最大获益方案如下:----------"<<endl;
DFS(0);
cout<<"选中的实验编号:"<<endl;
for(int i=1;i<=m;i++)
if(flag[i])
cout<<i<<" ";
cout<<endl;
cout<<"选中的仪器编号:"<<endl;
for(int i=m+1;i<=m+n;i++)
if(flag[i])
cout<<i-m<<" ";
}
int main()
{
int n, m,sum=0,total;
int cost,num;
memset(flag, 0, sizeof(flag));
cout<<"请输入实验数m和仪器数n:"<<endl;
cin>>m>>n;
init();
total=m+n;
cout<<"请依次输入实验产生的效益和该实验需要的仪器编号(为0结束):"<<endl;
for(int i=1;i<=m;i++)
{
cin>>cost;
sum+=cost;
add(0, i, cost); //源点到实验项目的边,容量为该项目效益
while(cin>>num,num) //num为该项目需要的仪器编号
add(i, m+num, INF); //实验项目到需要仪器的边,容量为无穷大
}
cout<<"请依次输入所有仪器的费用:"<<endl;
for(int j=m+1;j<=total;j++)
{
cin>>cost;
add(j, total+1, cost); //实验仪器到汇点的边,容量为实验仪器费用
}
cout<<endl;
printg(total+2);//输出初始网络邻接表
cout<<"最大净收益:"<<sum-Isap(0,total+1,total+2)<<endl;
cout<<endl;
printg(total+2);//输出最终网络邻接表
print(m,n); //输出最佳方案
return 0;
}
算法实现和测试
(1)运行环境
Code::Blocks
(2)输入
请输入实验数m和仪器数n:
5 15
请依次输入实验产生的效益和该实验需要的仪器编号(为0结束):
20 2 4 8 11 0
38 1 5 14 0
25 2 5 7 15 0
17 1 3 6 9 13 0
22 10 12 15 0
请依次输入所有仪器的费用:
2 7 4 8 10 1 3 7 5 9 15 6 12 17 8
(3)输出
最大净收益:23
----------最大获益方案如下:----------
选中的实验编号:
2 3 5
选中的仪器编号:
1 2 5 7 10 12 14 15