HDU 2868 Neighbor Friend

HDU 2868 Neighbor Friend

/**
HDU 2868 Neighbor Friend
题意：
有n个人，m对身高关系a,b(a的身高小于b的身高)
按身高排序，求有多少对人可以成为邻居
解题思路：
a,b,c三个人
a->b
a->c
b->c
a到c的最长距离大于1，所以a,c不能成为邻居
问题转化，找出不能成为邻居的个数{(a,c)}
剩下的就是邻居的可能数量{(a,b),(b,c)}
所以得到结果：2
解题方法两种：
dfs
floyd
*/

第一种：dfs方法

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
int dp[maxn];
vector<int> v[maxn];
int bo[maxn];
//dfs找到儿子节点最远距离
void dfs(int num,int ord){
    if(bo[num]>=2) return ;
    bo[num]=max(ord,bo[num]);
    for(int i=0;i<v[num].size();i++)
        dfs(v[num][i],ord+1);
}
int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        int a,b;
        for(int i=1;i<=n;i++) v[i].clear();
        for(int i=0;i<m;i++){
            scanf("%d%d",&a,&b);
            v[a].push_back(b);
        }
        int ans=0;
        for(int i=1;i<=n;i++){
            memset(bo,0,sizeof(bo));
            dfs(i,0);
            for(int j=1;j<=n;j++)
                if(bo[j]>=2) ans++;
        }
        ans=(n*(n-1))/2-ans;
        printf("%d\n",ans);
    }
    return 0;
}

第二种：floyd方法

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#define maxn 205
using namespace std;
int ma[maxn][maxn];
int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        int a,b;
        memset(ma,0,sizeof(ma));
        for(int i=0;i<m;i++){
            scanf("%d%d",&a,&b);
            ma[a][b]=1;
        }
        //弗洛伊德找到距离大于2的
        for(int k=1;k<=n;k++)
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                    if(ma[i][k]&&ma[k][j])
                        ma[i][j]=2;
        int ans=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                if(ma[i][j]<2&&ma[j][i]<2) ans++;
        printf("%d\n",ans);
    }
    return 0;
}