PTA_排座位（C语言）

布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。

输入格式：

输入第一行给出3个正整数：N（≤100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：宾客1 宾客2 关系，其中关系为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。
这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

输出格式：

对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出No problem；如果他们之间并不是朋友，但也不敌对，则输出OK；如果他们之间有敌对，然而也有共同的朋友，则输出OK but…；如果他们之间只有敌对关系，则输出No way。
输入样例：

7 8 4
5 6 1
2 7 -1
1 3 1
3 4 1
6 7 -1
1 2 1
1 4 1
2 3 -1
3 4
5 7
2 3
7 2

输出样例：

No problem
OK
OK but...
No way

**分析：**朋友的朋友也是朋友，可以通过判断两点是否连通来判断两人是否是朋友关系，并再次判断两人的直接关系，确定打印结果。第一次用广搜，判断两点是否连通，每判断一次就用一次搜索，结果正确但是超时了。后来决定用Floyd算法建立一个关系二维数组flag，判断两点是否能通过间接连通。输出时候直接通过数组索引。

#include <stdio.h>
#include <stdlib.h>

#define MAX 120
#define M 10000

typedef struct
{
    int n;
    int e;

    int edges[MAX][MAX];
}graph;

int flag[MAX][MAX];

void look_up_s(graph *g,int flag[MAX][MAX])
{
    int i,j,k;
    for (i=1;i<=g->n;i++)
    {
        for(j=0;j<g->n;j++)
        {
            if(g->edges[i][j]==1)
                flag[i][j]=1;
            else
                flag[i][j]=0;
        }
    }
    for(k=1;k<=g->n;k++)
    {
        for(i=1;i<=g->n;i++)
        {
            for (j=1;j<=g->n;j++)
            {
                if(flag[i][k]==1&&flag[k][j]==1)
                {
                    flag[i][j]=1;
                    flag[j][i]=1;
                }
            }
        }
    }

}




int main()
{
    graph *g=(graph *)malloc(sizeof(graph));
    int i,j,k,l,x;
    int f_d[2][M];
    scanf("%d%d%d",&g->n,&g->e,&x);
    for (i=1;i<=g->n;i++)
    {
        for(j=1;j<=g->n;j++)
        {
            g->edges[i][j]=0;
        }
    }
    for(i=1;i<=g->e;i++)
    {
        scanf("%d%d%d",&j,&k,&l);
        g->edges[j][k]=l;
        g->edges[k][j]=l;
    }
    for(i=0;i<x;i++)
    {
        scanf("%d%d",&f_d[0][i],&f_d[1][i]);
    }

    look_up_s(g,flag);

    for (i=0;i<x;i++)
    {
        if((g->edges[f_d[0][i]][f_d[1][i]]==1)||(g->edges[f_d[0][i]][f_d[1][i]]==0&&flag[f_d[0][i]][f_d[1][i]]==1))
        {
            printf("No problem\n");
            continue ;
        }
        if(g->edges[f_d[0][i]][f_d[1][i]]==-1&&flag[f_d[0][i]][f_d[1][i]]==1)
        {
            printf("OK but...\n");
            continue;
        }
        if(g->edges[f_d[0][i]][f_d[1][i]]==-1&&flag[f_d[0][i]][f_d[1][i]]==0)
        {
            printf("No way\n");
            continue ;
        }
        if(g->edges[f_d[0][i]][f_d[1][i]]==0&&flag[f_d[0][i]][f_d[1][i]]==0)
        {
            printf("OK\n");
            continue ;
        }
    }

    return 0;
}