poj 2175--指派问题

题意：

n幢大楼，m个防空洞，给出每个大楼的人数和防空洞可容纳人数，以及大楼到防空洞费用的计算方式，给定一个方案，问是否总的费用最优（最小），若不是最优，给出一个更优的解。

思路：

典型的指派类问题，大楼和防空洞分别做点，并新增源点s,汇点t，s向每幢大楼连一条容量为大楼人数，费用为0的边，每幢大楼向每个防空洞连一条容量为inf ，费用为相应费用的边，每个防空洞向t点连一条容量为防空洞容量，费用为0的边。如此可得一二分图G。此题本可以在G上求解流量为大楼中人数总和的最下费用流来求得最优解。时间复杂度为O(F|e|logv)，对于此题来说会超时。
由最小费用流的性质可知：流fi为最小费用流，当且仅当流fi的残余网络中没有费用负环。
由于题中不需要求得最优解，那么我们只需要按照题给出的方案去增广，然后在其残余网络上寻找负环。沿着某一负环增广1的流量，就可以得到一个更优的解。此处可以用spfa来寻找负环。

个人注意：

在spfa中，若某一个顶点入队次数大于等于n（总的节点数），那么源点到此点的最短路径上至少存在一个负环（此点可能并不在负环上）。
求解图中任一负环，可以新增源点，并向图中所有点增加费用为0的边，然后以此源点求解。

代码：

#include <cstdio>
#include <vector>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn = 250;
const int inf = 2e9;
struct edge{
    int to,cap,cost;
    edge()
    {
        cap = 0;
        to = -1;
        cost = 0;
    }
};
struct point{int x,y,c;};
point px[maxn];
edge G[maxn][maxn];
int prev[maxn];
int dis[maxn];
bool inqueue[maxn];
int cnt[maxn];
bool used[maxn];//找出负环
int n,m;
int now[maxn][maxn];
queue <int> que;
int inline cal(int i,int j)
{
    return abs(px[i].x - px[j].x)+abs(px[i].y - px[j].y)+1;
}
void add_edge(int from,int to,int cap,int cost)
{
    G[from][to].to = to;
    G[from][to].cap = cap;
    G[from][to].cost = cost;
    G[to][from].to = from;
    G[to][from].cap = 0;
    G[to][from].cost = -cost;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 1; i <= n; i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        px[i].x = a;
        px[i].y = b;
        px[i].c = c;
        add_edge(0,i,c,0);
        add_edge(n+m+2,i,inf,0);
    }
    for(int i = 1; i <= m; i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        px[i+n].x = a;
        px[i+n].y = b;
        px[i+n].c = c;
        for(int j = 1; j<=n ;j++)
            add_edge(j,i+n,min(px[j].c,c),cal(j,i+n));
        add_edge(i+n,n+m+1,c,0);
        add_edge(n+m+2,i+n,inf,0);
    }
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
        {
            int a;
            scanf("%d",&a);
            now[i][j] = a;
            G[i][j+n].cap -=a;
            G[j+n][i].cap +=a;
            G[0][i].cap-=a;
            G[i][0].cap+=a;
            G[j+n][n+m+1].cap-=a;
            G[n+m+1][j+n].cap+=a;
        }
    add_edge(n+m+2,0,inf,0);
    add_edge(n+m+2,n+m+1,inf,0);

    bool is = false;//有无负环的标志

    fill(dis,dis+ n+m+3,inf);
    for(int i = 0; i < n+m+3;i++)
        prev[i] = i;

    inqueue[n+m+2] = true;
    dis[n+m+2] = 0;
    que.push(n+m+2);
    cnt[n+m+2] = 1;
    while(!que.empty())
    {
        int son = que.front();que.pop();
        inqueue[son] = false;
        if(inqueue[prev[son]]) continue;
        for(int i = 0; i < n+m+3;i++)
        {
            if(G[son][i].cap > 0  && dis[i] > dis[son] + G[son][i].cost)
            {
                dis[i] = dis[son] + G[son][i].cost;
                prev[i] = son;
                if(!inqueue[i])
                {
                    inqueue[i] = true;
                    que.push(i);
                    cnt[i]++;
                    if(cnt[i] >= n+m+3)
                    {
                        is = true;
                        memset(used, false,sizeof(used));
                        int x;
                        for(x = i;!used[x];x = prev[x])
                            used[x] = true;
                        bool fs = false;//第一次进入s
                        for(int j = x; !fs||j!=x; j = prev[j])
                        {
                            fs = true;
                            int a = prev[j],b = j;
                            if(a > 0 && a <= n && b > n && b <= n+m)
                            {
                                now[a][b - n]++;
                            }else if(b > 0 && b <=n && a > n && a <= n+m)
                            {
                                now[b][a - n]--;
                            }
                        }
                        printf("SUBOPTIMAL\n");
                        for(int j = 1;j <= n;j++)
                            for(int k =1; k <= m; k++)
                            {
                                printf("%d",now[j][k]);
                                if(k== m)
                                    printf("\n");
                                else
                                    printf(" ");
                            }
                        break;
                    }
                }
            }
        }
        if(is)
            break;
    }
    if(!is)
        printf("OPTIMAL\n");
    return 0;
}