xmu 1353: 攻占城堡

这题是厦门大学第一届程序设计大赛上的一题,这题我用的是prim算法.稍微有一点变形;

说实话,可能不和我说是最小生成树解法,我可能想不到,搞不好要去暴力..

虽然最后还是想出来了...其实不难..

将读入的每个城堡的分值读入,读入以后计算城堡i到城堡j的分值,平方并且记录.

最后将最小生成树改成最大生成树..随便叫= =反正是这个意思..

其实感觉有点像以前做过的一道DP题,不知道可不可行..hdu 的 搬寝室.

#include <iostream>
#include <string.h>
using namespace std;
#define maxn 1005
#define  INF 0x3f3f3f3f
int mincost[maxn],value[maxn];
int cost[maxn][maxn];
bool visit[maxn];
int n;
int prim()
{
    int res = 0, index = 1 , max_cost ;
    for(int i = 1;i <= n;i++)
        mincost[i] = cost[i][1];
    mincost[1] = 0;
    memset(visit,false,sizeof(visit));
    visit[1] = true;
    for(int i = 1 ; i <= n; i++)
    {
        max_cost = 0;
        for(int j = 1; j <= n; j++)
        {
            if(mincost[j]>max_cost && !visit[j])
            {
                max_cost = mincost[j];
                index = j;
            }
        }
        visit[index] = true;
        for(int k = 1; k <= n; k++)
        {
            if(!visit[k] && cost[k][index] > mincost[k])
                mincost[k] = cost[k][index] ;

        }

    }
    for(int i = 1 ;i <=n;i++)
        res += mincost[i];
    return res;
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(cost,0,sizeof(cost));
        for(int i = 1 ; i <= n ; i ++)
        {
            scanf("%d",&value[i]);
            for(int j = 1 ; j <= i ;j++)
            {
                cost[i][j] = cost[j][i] = (value[i] - value[j])*(value[i]-value[j]); 
            }
        }
        printf("%d
",prim());
    }
    return 0 ;
}