Codeforces Zepto Code Rush 2014 -C - Dungeons and Candies

这题给的一个教训：Codeforces没有超时这个概念。本来以为1000*(1000+1)/2*10*10要超时的。结果我想多了。

这题由于k层都可能有关系，所以建一个图，每两个点之间连边，边权为n*m和他们之间的差值*w的最小值，然后求一个最小生成树就可以得出结果。且可以证明不会存在环。由于边比较稠密，用Prim算法求最小生成树。

代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

#include <algorithm>

#define Mod 1000000007

using namespace std;

#define N 2007



char mp[1006][12][12];

int G[1006][1005],from[1006];

int n,m,res,id,k,w;

int vis[1005],len[1005];



struct Ans

{

    int ind,pre;

}ans[1007];



int min(int ka,int kb)

{

    if(ka < kb)

        return ka;

    return kb;

}



int dif(char a[12][12],char b[12][12])

{

    int cnt = 0;

    for(int i=0;i<n;i++)

    {

        for(int j=0;j<m;j++)

        {

            if(a[i][j] != b[i][j])

                cnt++;

        }

    }

    return cnt;

}



void Prim()

{

    int i,j,tag,mini;

    res = 0;

    id = 0;

    memset(vis,0,sizeof(vis));

    memset(from,0,sizeof(from));

    for(i=1;i<=k;i++)

        len[i] = n*m;   //最大为n*m

    for(i=1;i<=k;i++)

    {

        mini = Mod;

        for(j=1;j<=k;j++)

        {

            if(!vis[j] && len[j] < mini)

            {

                mini = len[j];

                tag = j;

            }

        }

        if(mini == Mod)

            return;

        res += len[tag];

        if(len[tag] == n*m)

            ans[id].ind = tag,ans[id++].pre = 0;

        else

            ans[id].ind = tag,ans[id++].pre = from[tag];

        vis[tag] = 1;

        for(j=1;j<=k;j++)

        {

            if(!vis[j] && len[j] > G[tag][j])

            {

                len[j] = G[tag][j];

                from[j] = tag;

            }

        }

    }

}



int main()

{

    int i,j;

    while(scanf("%d%d%d%d",&n,&m,&k,&w)!=EOF)

    {

        id = 0;

        for(i=1;i<=k;i++)

        {

            for(j=0;j<n;j++)

                scanf("%s",mp[i][j]);

        }

        int T = n*m;

        for(i=1;i<=k;i++)

        {

            for(j=i+1;j<=k;j++)

                G[j][i] = G[i][j] = min(dif(mp[i],mp[j])*w,T);

            G[i][i] = 0;

        }

        Prim();

        printf("%d\n",res);

        for(i=0;i<id;i++)

            printf("%d %d\n",ans[i].ind,ans[i].pre);

    }

    return 0;

}

View Code