HDU 5253 连接的管道 【kruskal】【TLE原因总结】

连接的管道

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 7251    Accepted Submission(s): 2230

 

Problem Description

老 Jack 有一片农田，以往几年都是靠天吃饭的。但是今年老天格外的不开眼，大旱。所以老 Jack 决定用管道将他的所有相邻的农田全部都串联起来，这样他就可以从远处引水过来进行灌溉了。当老 Jack 买完所有铺设在每块农田内部的管道的时候，老 Jack 遇到了新的难题，因为每一块农田的地势高度都不同，所以要想将两块农田的管道链接，老 Jack 就需要额外再购进跟这两块农田高度差相等长度的管道。

现在给出老 Jack农田的数据，你需要告诉老 Jack 在保证所有农田全部可连通灌溉的情况下，最少还需要再购进多长的管道。另外，每块农田都是方形等大的，一块农田只能跟它上下左右四块相邻的农田相连通。

 

Input

第一行输入一个数字T(T≤10)，代表输入的样例组数

输入包含若干组测试数据，处理到文件结束。每组测试数据占若干行，第一行两个正整数 N,M(1≤N,M≤1000)，代表老 Jack 有N行*M列个农田。接下来 N 行，每行 M 个数字，代表每块农田的高度，农田的高度不会超过100。数字之间用空格分隔。

 

Output

对于每组测试数据输出两行：

第一行输出："Case #i:"。i代表第i组测试数据。

第二行输出 1 个正整数，代表老 Jack 额外最少购进管道的长度。

 

Sample Input

2 4 3 9 12 4 7 8 56 32 32 43 21 12 12 2 3 34 56 56 12 23 4

 

Sample Output

Case #1: 82 Case #2: 74

 

Source

2015年百度之星程序设计大赛 - 初赛(2)

 

本题的思路：

很明显的最小生成树，只是自己可能解出少了，没反应过来。

一开始超时的原因：超时的原因真离谱

e[]数组的下标范围开的太小。开的4*N。显然并不对，并没有真正去算

精确算的话应该是(n-1)*(m)+(m-1)*n

大概是 2*N*N

 

#include <bits/stdc++.h>
using namespace std;
const int N=1010,M=N*N;
int g[N][N];
struct Node
{
   int u,v;
   int w; 
   
   bool operator<(const Node &W)const
   {
       return w<W.w;
   }
}e[2*M];
int p[M];

int dx[3]={0,1};
int dy[3]={1,0};

int find(int x)
{
    if(p[x]!=x)
    {
        p[x]=find(p[x]);
    }
    return p[x];
}


int main()
{
    int T;
    scanf("%d",&T);
    for(int t=1;t<=T;t++)
	{
		printf("Case #%d:\n",t); 
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
        {
            scanf("%d",&g[i][j]);
        }
        
        for(int i=0;i<M;i++)
        p[i]=i;
        
        int idx=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                int t=g[i][j];
                for(int k=0;k<2;k++)
                {
                    int x=i+dx[k];
                    int y=j+dy[k];
                    if(x<0||x>=n||y<0||y>=m) continue;
                    e[idx++]={i*m+j,x*m+y,abs(t-g[x][y])};
                }
            }
        }
        
        
        sort(e,e+idx);
        /*
        for(int i=0;i<idx;i++)
        {
            cout<<e[i].w<<" "; 
        }
        */
        
        int cnt=0,ans=0;
        for(int i=0;i<idx;i++)
        {
            int u=e[i].u;
            int v=e[i].v;
            int w=e[i].w;
            
            u=find(u);
            v=find(v);
            
            if(u!=v)
            {
                p[u]=v;
                ans+=w;
                cnt++;
            }
        }
        
        printf("%d\n",ans);
    }
    
    return 0;
}