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

#include <iostream>
#include <string.h>
#include <map>
#include <set>
#include <vector>
#include <stdio.h>
#include <algorithm>
using namespace std;
int dp[55][55];
char s[55];

void solve()
{
    int T;
    cin>>T;
    while(T--)
    {
        int n,d;
        scanf("%d %d",&n,&d);
        memset(dp,0x3f,sizeof dp);
        for(int i = 0; i < n; ++i)
        {
            scanf("%s",s);
            for(int j = 0; j < n; ++j)
            {
                if(s[j] == 'Y')
                {
                    dp[i][j] = d;
                }
            }
        }
        for(int k = 0; k < n; ++k)
        {
            for(int i = 0; i < n; ++i)
            {
                for(int j = 0; j < n; ++j)
                {
                    dp[i][j] = min(dp[i][k] + dp[k][j],dp[i][j]);
                }
            }
        }
        int ans = 0;
        bool ok = true;
        for(int i = 0; i < n; ++i)
        {
            for(int j = i + 1; j < n; ++j)
            {
                if(dp[i][j] != (0x3f3f3f3f))
                {
                    ans = max(dp[i][j],ans);
                    ok = false;
                }
                else
                {
                    ok = true;
                    goto S;
                }
            }
        }
        S: if(ok) ans = -1;
        printf("%d\n",ans);
    }
}

int main() {
    solve();
    return 0;
}