51Nod 1366 贫富差距

1366 贫富差距 

题目来源： TopCoder

基准时间限制：1 秒 空间限制：131072 KB 分值: 40 难度：4级算法题

一个国家有N个公民，标记为0,1,2，...，N-1，每个公民有一个存款额。已知每个公民有一些朋友，同时国家有一条规定朋友间的存款额之差不能大于d。也就是说，a和b是朋友的话，a有x元的存款，b有y元，那么|x-y|<=d。给定d值与N个人的朋友关系，求这个国家最富有的人和最贫穷的人的存款相差最大的可能值是多少？即求贫富差距的最大值的下界。若这个值为无穷大，输出-1.

Input

多组测试数据，第一行一个整数T，表示测试数据数量，1<=T<=5
每组测试数据有相同的结构构成。
每组数据的第一行两个整数N，d，表示人数与朋友间存款差的最大值，其中2<=N<=50,0<=d<=1000.
接下来有一个N*N的数组A，若A[i][j]='Y'表示i与j两个人是朋友，否则A[i][j]='N'表示不是朋友。其中A[i][i]='N'，且保证
A[i][j]=A[j][i].

Output

每组数据一行输出，即这个国家的贫富差距最大值的下界，如果这个值为无穷大输出-1.

Input示例

3
3 10
NYN
YNY
NYN
2 1
NN
NN
6 1000
NNYNNN
NNYNNN
YYNYNN
NNYNYY
NNNYNN
NNNYNN

Output示例

20
-1
3000

题解：Floyd跑最短路，是朋友的点距离为1，不是的为inf，如果存在两个点直接的点距是inf说明有多个连通图，那么输出-1，否则输出各点距的最大值*D。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 55;
#define inf 0x3f3f3f
int a[maxn][maxn];
int n,m,t;
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
char s;
cin>>s;
if(i==j)a[i][j]=0;
else if(s=='N')a[i][j]=inf;
else a[i][j]=1;
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
for(int k=1;k<=n;k++)
{
a[j][k]=min(a[j][k],a[j][i]+a[i][k]);
}
}
}
int ans = 0;
int ok = 0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(a[i][j]==inf)ok=1;
else ans = max(ans,a[i][j]);
}
}
if(ok==1)printf("-1\n");
else printf("%d\n",ans*m);
}
return 0;
}