★★ZOJ 2342 KM算法+不等式

题意：给你n个点，m条边，前n-1条边可以构成一个生成树，每条边都有一个权值，现在让你调整权值，使得前n-1条边构成这幅图的最小生成树，所有|ci-di|的和最小。

分析：首先我们可以发现，前n-1条边要构成最小生成树，那么这些边的权值只能减小，后n~m的边只能增大，我们先将前n-1条边构成一棵生成树，然后对于后面的n~m的边，假设任意这些边中的一条为j，则加入这条边就会产生环，为了一开始这条j边不能加入生成树，则这个环里面的任意一条边的权值要<=j。假设改变后的权值表示为ci-li，cj+lj，则有ci-li<=cj+lj,即为li+lj>=ci-cj，设ci-cj为wij,则为li+lj>=wij，到了这里有没有感觉就是KM匹配应该满足的顶标条件呢？其实我们要的是它满足条件的最小权值，转化为KM的最大权值匹配，最后顶标数组就是一组解。

代码：

#pragma comment(linker,"/STACK:102400000,102400000")
#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
using namespace std;
typedef long long ll;
const int maxn=505;
const int maxm=1000005;
const int INF=1000000000;
struct EdgeNode{int from,to,id,next;}edge[maxm];
int head[maxn],cnt,cost[maxn],vis[maxn];
void add(int x,int y,int id)
{
    edge[cnt].from=x;
    edge[cnt].to=y;
    edge[cnt].id=id;
    edge[cnt].next=head[x];
    head[x]=cnt++;
}
void init()
{
    cnt=0;
    memset(head,-1,sizeof(head));
}
int n,nx,ny,m;
int march[maxn],lx[maxn],ly[maxn],slack[maxn];
int visx[maxn],visy[maxn],w[maxn][maxn];

int dfs(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])
            continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0)
        {
            visy[y]=1;
            if(march[y]==-1||dfs(march[y]))
            {
                march[y]=x;
                return 1;
            }
        }
        else if(slack[y]>t)
            slack[y]=t;
    }
    return 0;
}

int KM()
{
    int i,j;
    memset(march,-1,sizeof(march));
    memset(ly,0,sizeof(ly));
    for(i=1;i<=nx;i++)
        for(j=1,lx[i]=-INF;j<=ny;j++)
            if(w[i][j]>lx[i])
               lx[i]=w[i][j];
    for(int x=1;x<=nx;x++)
    {
        for(i=1;i<=ny;i++)slack[i]=INF;
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(x))break;
            int d=INF;
            for(i=1;i<=ny;i++)
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            for(i=1;i<=nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(i=1;i<=ny;i++)
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
        }
    }
    int res=0;
    for(i=1;i<=ny;i++)
        if(march[i]>-1)
            res+=w[march[i]][i];
    return res;
}

bool find(int x,int y,int id)
{
    if(x==y)return 1;
    vis[x]=1;
    for(int i=head[x];i!=-1;i=edge[i].next)
    {
        int v=edge[i].to;
        if(vis[v])continue;
        if(find(v,y,id)){
            if(cost[edge[i].id]>cost[id])
               w[edge[i].id][id]=cost[edge[i].id]-cost[id];
            return 1;
        }
    }
    return 0;
}

int main()
{
    int x,y,z,id,i,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(i=1;i<n;i++){
            scanf("%d%d%d",&x,&y,&cost[i]);
            add(x,y,i);
            add(y,x,i);
        }
        memset(w,0,sizeof(w));
        for(i=n;i<=m;i++){
            scanf("%d%d%d",&x,&y,&cost[i]);
            memset(vis,0,sizeof(vis));
            find(x,y,i);
        }
        nx=n-1; ny=m;
        KM();
        for(i=1;i<n;i++)
        printf("%d\n",cost[i]-lx[i]);
        for(i=n;i<=m;i++)
        printf("%d\n",cost[i]+ly[i]);
    }
    return 0;
}