首页 > 代码库 > HDU4126Genghis Khan the Conqueror(最小生成树+并查集)

HDU4126Genghis Khan the Conqueror(最小生成树+并查集)

Genghis Khan the Conqueror

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 327680/327680 K (Java/Others)
Total Submission(s): 1687    Accepted Submission(s): 501


Problem Description
Genghis Khan(成吉思汗)(1162-1227), also known by his birth name Temujin(铁木真) and temple name Taizu(元太祖), was the founder of the Mongol Empire and the greatest conqueror in Chinese history. After uniting many of the nomadic tribes on the Mongolian steppe, Genghis Khan founded a strong cavalry equipped by irony discipline, sabers and powder, and he became to the most fearsome conqueror in the history. He stretched the empire that resulted in the conquest of most of Eurasia. The following figure (origin: Wikipedia) shows the territory of Mongol Empire at that time.
技术分享

Our story is about Jebei Noyan(哲别), who was one of the most famous generals in Genghis Khan’s cavalry. Once his led the advance troop to invade a country named Pushtuar. The knights rolled up all the cities in Pushtuar rapidly. As Jebei Noyan’s advance troop did not have enough soldiers, the conquest was temporary and vulnerable and he was waiting for the Genghis Khan’s reinforce. At the meantime, Jebei Noyan needed to set up many guarders on the road of the country in order to guarantee that his troop in each city can send and receive messages safely and promptly through those roads.

There were N cities in Pushtuar and there were bidirectional roads connecting cities. If Jebei set up guarders on a road, it was totally safe to deliver messages between the two cities connected by the road. However setting up guarders on different road took different cost based on the distance, road condition and the residual armed power nearby. Jebei had known the cost of setting up guarders on each road. He wanted to guarantee that each two cities can safely deliver messages either directly or indirectly and the total cost was minimal.

Things will always get a little bit harder. As a sophisticated general, Jebei predicted that there would be one uprising happening in the country sooner or later which might increase the cost (setting up guarders) on exactly ONE road. Nevertheless he did not know which road would be affected, but only got the information of some suspicious road cost changes. We assumed that the probability of each suspicious case was the same. Since that after the uprising happened, the plan of guarder setting should be rearranged to achieve the minimal cost, Jebei Noyan wanted to know the new expected minimal total cost immediately based on current information.
 

Input
There are no more than 20 test cases in the input.
For each test case, the first line contains two integers N and M (1<=N<=3000, 0<=M<=N×N), demonstrating the number of cities and roads in Pushtuar. Cities are numbered from 0 to N-1. In the each of the following M lines, there are three integers xi, yi and ci(ci<=107), showing that there is a bidirectional road between xi and yi, while the cost of setting up guarders on this road is ci. We guarantee that the graph is connected. The total cost of the graph is less or equal to 109.

The next line contains an integer Q (1<=Q<=10000) representing the number of suspicious road cost changes. In the following Q lines, each line contains three integers Xi, Yi and Ci showing that the cost of road (Xi, Yi) may change to Ci (Ci<=107). We guarantee that the road always exists and Ci is larger than the original cost (we guarantee that there is at most one road connecting two cities directly). Please note that the probability of each suspicious road cost change is the same.
 

Output
For each test case, output a real number demonstrating the expected minimal total cost. The result should be rounded to 4 digits after decimal point.
 

Sample Input
3 3 0 1 3 0 2 2 1 2 5 3 0 2 3 1 2 6 0 1 6 0 0
 

Sample Output
6.0000
Hint
The initial minimal cost is 5 by connecting city 0 to 1 and city 0 to 2. In the first suspicious case, the minimal total cost is increased to 6; the second case remains 5; the third case is increased to 7. As the result, the expected cost is (5+6+7)/3 = 6.
 

Source
2011 Asia Fuzhou Regional Contest
 
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;

const int N = 3005;
const double inf = 0x3f3f3f3f;
struct EDG
{
    int u,v;
    double c;
};
struct TO
{
    int v;
    double c;
};

vector<TO>tmap[N];
EDG edg[N];
int n,treeEdg[N][N];
double node[N];
EDG tedg[N*N];
int fath[N];

int cmp(EDG a,EDG b)
{
    return a.c<b.c;
}
int findfath(int x)
{
    if(x==fath[x])
        return fath[x];
    fath[x]=findfath(fath[x]);
    return fath[x];
}
double MST(int m)
{
    double sum=0;
    int k=0;
    for(int i=0; i<m; i++)
    {
        int x=findfath(tedg[i].u);
        int y=findfath(tedg[i].v);
        if(x!=y)
        {
            k++;
            treeEdg[tedg[i].u][tedg[i].v]=treeEdg[tedg[i].v][tedg[i].u]=k;
            edg[k].u=tedg[i].u; edg[k].v=tedg[i].v; edg[k].c=tedg[i].c;
            fath[x]=y; sum+=tedg[i].c;
            if(k==n-1)
                break;
        }
    }
    return sum;
}
int main()
{
    int m,q,a,b;
    double c,ans,sum,tc;
    TO ss;

    while(scanf("%d%d",&n,&m)>0&&n+m!=0)
    {
        for(int i=0;i<=n;i++)
        {
            fath[i]=i;
            for(int j=0;j<=n;j++)
                treeEdg[i][j]=N;
        }

        for(int i=0;i<m;i++)
        {
            scanf("%d%d%lf",&a,&b,&c);
            tedg[i].u=a;
            tedg[i].v=b;
            tedg[i].c=c;
        }
        sort(tedg,tedg+m,cmp);
        sum=MST(m);
        scanf("%d",&q);
        ans=0;
        for(int j=0;j<q;j++)
        {
            scanf("%d%d%lf",&a,&b,&c);

            if(treeEdg[a][b]==N)
                ans+=sum;
            else
            {
                for(int i=0;i<=n;i++)
                    fath[i]=i;
                for(int i=1;i<n;i++)
                {
                    if(treeEdg[a][b]==i)
                    {
                       tc=edg[i].c; continue;
                    }
                    int x=findfath(edg[i].u);
                    int y=findfath(edg[i].v);
                    fath[x]=y;
                }
                int flag=0;
                for(int i=0;i<m&&tedg[i].c<c;i++)
                {
                    if(treeEdg[tedg[i].u][tedg[i].v]!=N)
                        continue;

                        int x=findfath(tedg[i].u);
                        int y=findfath(tedg[i].v);
                        fath[x]=y;
                         x=findfath(a);
                         y=findfath(b);
                        if(x==y)
                        {
                            ans=ans+sum-tc+tedg[i].c; flag=1; break;
                        }
                }
                if(flag==0)
                    ans=ans+sum-tc+c;
            }
        }
        printf("%.4lf\n",ans/(q*1.0));
    }
}


HDU4126Genghis Khan the Conqueror(最小生成树+并查集)