poj 1639 Picnic Planning 度限制mst

https://vjudge.net/problem/POJ-1639

题意：

有一群人，他们要去某一个地方，每个车可以装无数个人，给出了n条路，包含的信息有路连接的地方，以及路的长度，路是双向的，但是终点只有一个，并且终点能停的车的数量是有限制的，问最少走的路是多少。

思路：

因为终点的停车的数量是有限制的，所以终点的度是有限制的，又因为这题可以用最小生成树解决，所以就是度限制最小生成树。

度限制最小生成树求解思想并不复杂，首先我们把有度限制的点给忽略，然后给每一个连通分量求最小生成树，最后把每一个连通分量中与有度限制的点的距离最小的点与度限制点连接，假设有m个连通分量。

那么我们现在求出了m限制的最小生成树，假如限制数k < m，那么就无解。

当k >= m时，我们可以在m度限制mst的基础上，求m + 1，m + 2。。。k度限制最小生成树，求法也不是很难懂，但是程序就很难写了Orz。

如何求呢？枚举每一条未在生成树中与（现在我们把度限制点叫做R点）R点相连的边，然后把边加入生成树，必然会形成环，然后把环中与R点不相连的权值最大的边去掉，枚举之后的最小值就是m+1度限制最小生成树的值。然后依次求到k限制mst，求其中的最小值。

但是，依次枚举的话时间复杂度非常高，所以我们要优化。这时就用到了动态规划的思想。将与R点到其它点的边权值最大求出，之后加边的时候，直接替换就可以了。

转移方程 dp[v] = max(dp[father(v)],w(v , father(v)));

看不懂就多看几遍Orrrrrrrrrrrrrrrrrrrrrrrrz。

代码：

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <map>
#include <string>
using namespace std;

const int inf = 0x3f3f3f3f;

struct edge
{
    int x,y;
    int v;
} a[5055],dp[5055];

map<string,int> mmp;
bool flag[105][105];
int par[105];
int g[105][105];

int ans;
int num;
int du,lim;

int fin(int x)
{
    if (x == par[x]) return x;
    else return par[x] = fin(par[x]);
}

void unit(int x,int y)
{
    x = fin(x);
    y = fin(y);

    if (x == y) return;

    par[x] = y;
}

void dfs(int cur,int pre)
{
    for (int i = 2;i <= num;i++)
    {
        if (i != pre && flag[cur][i])
        {
            if (dp[i].v == -1)
            {
                if (dp[cur].v > g[cur][i])
                {
                    dp[i] = dp[cur];
                }
                else
                {
                    dp[i].x = cur;
                    dp[i].y = i;
                    dp[i].v = g[cur][i];
                }
            }

            dfs(i,cur);
        }
    }
}

void solve(void)
{
    for (int i = du + 1;i <= lim;i++)
    {
        memset(dp,-1,sizeof(dp));

        dp[1].v = -inf;

        for (int j = 2;j <= num;j++)
            if (flag[j][1]) dp[j].v = -inf;

        dfs(1,-1);

        int mi = inf,tmp;

        for (int j = 2;j <= num;j++)
        {
            if (g[1][j] != -1)
            {
                if (mi > g[1][j] - dp[j].v)
                {
                    mi = g[1][j] - dp[j].v;
                    tmp = j;
                }
            }
        }

        if (mi >= 0) break;

        ans += mi;

        int x = dp[tmp].x,y = dp[tmp].y;

        flag[x][y] = flag[y][x] = 0;

        flag[1][tmp] = flag[tmp][1] = 1;
    }
}

int get_num(string aa)
{
    if (mmp[aa]) return mmp[aa];
    else
    {
        mmp[aa] = ++num;
        return num;
    }
}

bool cmp(edge aa,edge bb)
{
    return aa.v < bb.v;
}

int main()
{
    num = 1;

    mmp["Park"] = 1;

    memset(g,-1,sizeof(g));

    int n;

    scanf("%d",&n);

    for (int i = 0;i < n;i++)
    {
        string aa,bb;
        int v;

        cin >> aa >> bb;

        scanf("%d",&v);

        int x = get_num(aa),y = get_num(bb);

        if (g[x][y] == -1) g[x][y] = g[y][x] = v;
        else g[x][y] = g[y][x] = min(v,g[x][y]);

        a[i].x = x;
        a[i].y = y;
        a[i].v = g[x][y];
    }

    for (int i = 0;i <= num;i++) par[i] = i;

    scanf("%d",&lim);

    sort(a,a+n,cmp);

    for (int i = 0;i < n;i++)
    {
        int x = a[i].x,y = a[i].y;

        if (x == 1 || y == 1) continue;
        if (fin(x) == fin(y)) continue;

        ans += a[i].v;

        unit(x,y);

        flag[x][y] = flag[y][x]  =1;
    }

    int minn[105],tmp[105];

    memset(minn,inf,sizeof(minn));

    for (int i = 2;i <= num;i++)
    {
        int rt = fin(i);

        if (g[1][i] != -1)
        {
            if (g[1][i] < minn[rt])
            {
                minn[rt] = g[1][i];
                tmp[rt] = i;
            }
        }
    }

    for (int i = 2;i <= num;i++)
    {
        if (minn[i] != inf)
        {
            du++;
            flag[1][tmp[i]] = flag[tmp[i]][1] = 1;
            ans += minn[i];
        }
    }

    solve();

    printf("Total miles driven: %d\n",ans);

    return 0;
}