POJ 1161 walls

 

 http://poj.org/problem?id=1161

floyd+枚举

把区域看成节点建图   

#include<stdio.h>
#include<string.h>
#include<vector>
#define N 255
#define M 255
#define INF 0xfffffff
using namespace std;

int m,n,tot;
int g[M][M];//区域i，j之间的距离 
int mber[40];//成员所在的城镇 
int a[N];
int dis[N][M];//dis[i][j]表是城镇i到区域j的最短距离，即所需穿过的墙最少 
int belong[N][N][2];//belong[i][j]存的是城镇i，j之间的边所属的两个区域 

int min(int a,int b)
{
	if(a<b) return a;
	return b;
}

void floyd()
{
	int i,j,k;
	for(k=1;k<=m;k++)
		for(i=1;i<=m;i++)
			for(j=1;j<=m;j++)
				if(g[i][j]>g[i][k]+g[k][j])
					g[i][j]=g[i][k]+g[k][j];
}

int Get_Min()
{
	int i,j,k;
	int mn=INF;
	for(i=0;i<=n;i++)
		for(j=0;j<=m;j++)
			dis[i][j]=INF;
	for(i=0;i<tot;i++)//枚举成员 
	{
		int u=mber[i];
		for(j=1;j<=n;j++)//枚举成员所在城镇所在的边 
		{
			if(belong[u][j][0]==0) continue;
			int v1=belong[u][j][0];//成员所在城镇所属的两个区域 v1 v2 
			int v2=belong[u][j][1];
			for(k=1;k<=m;k++)//成员所在城镇到区域k的最短距离 
				dis[u][k]=min(dis[u][k],min(g[v1][k],g[v2][k]));
		}
	}
	for(i=1;i<=m;i++)
	{
		int tmp=0;
		for(j=0;j<tot;j++)
		{
			int u=mber[j];
			tmp+=dis[u][i];
		}
		if(tmp<mn) 
			mn=tmp;
	}
	return mn;
}

int main()
{
	int i,j;
	while(scanf("%d",&m)!=EOF)
	{
		for(i=0;i<=m+1;i++)
			for(j=0;j<=m+1;j++)
				g[i][j]=( i==j? 0:INF);
		memset(belong,0,sizeof(belong));
		scanf("%d%d",&n,&tot);
		for(i=0;i<tot;i++)
			scanf("%d",&mber[i]);
		for(i=1;i<=m;i++)
		{
			int num;
			scanf("%d",&num);
			for(j=0;j<num;j++)
				scanf("%d",&a[j]);
			for(j=0;j<num;j++)
			{
				int k=a[j];
				int l=a[(j+1)%num];
				if(belong[k][l][0]==0)
				{
					belong[k][l][0]=i;
					belong[l][k][0]=i;
				}
				else
				{
					int u=belong[k][l][0];
					g[i][u]=1; g[u][i]=1;
					belong[k][l][1]=i;
					belong[l][k][1]=i;
				}
			}
		}
		floyd();//求各个区域间的最短路 
		printf("%d\n",Get_Min());
	}
	return 0;
}