布置宴席最微妙的事情,就是给前来参宴的各位宾客安排座位。无论如何,总不能把两个死对头排到同一张宴会桌旁!这个艰巨任务现在就交给你,对任何一对客人,请编写程序告诉主人他们是否能被安排同席。
输入格式:
输入第一行给出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
解题思路:由题目可知,朋友的关系可以传递,而敌人的关系不能传递,所以对于朋友的关系采用并查集来确定,而敌人的关系采用二维数组来确定,最后针对判定的情况输出。
import java.util.Scanner;
public class Main {
static int friends[];//定义一个数组朋友,索引为某人的编号,值为其朋友的编号;
public static void main(String args[]) {
Scanner scanner=new Scanner(System.in);
int n=scanner.nextInt();
int m=scanner.nextInt();
int k=scanner.nextInt();
friends=new int[n+1];
for(int i=1;i<n+1;i++) {
friends[i]=i;//初始化数组,是每个人的朋友首先是自己;
}
int enemy[][]=new int[n+1][n+1];//存储两个人之间的敌对关系
for(int i=0;i<m;i++) {
int a=scanner.nextInt();
int b=scanner.nextInt();
int c=scanner.nextInt();
if(c==-1) {
enemy[a][b]=1;
enemy[b][a]=1;
}
else {
if(find(a)!=find(b))
union(a, b);
}
}
//记过上述的循环则可以通过查询friends与enemy数组查询任意两个人之间的关系;
for(int i=0;i<k;i++) {
int a=scanner.nextInt();
int b=scanner.nextInt();
if(find(a)!=find(b)) {
if(enemy[a][b]==1)
System.out.println("No way");
else
System.out.println("OK");
}
else {
if(enemy[a][b]==1)
System.out.println("OK but...");
else {
System.out.println("No problem");
}
}
}
}
/**
*
* @通过下面的两个类可以查明两位宾客是否有共同的朋友
*/
//寻找共同朋友
public static int find(int a) {
if(a==friends[a])
return a;
friends[a]=find(friends[a]);
return friends[a];
}
//合并节点
public static void union(int a,int b) {
friends[find(a)]=find(b);
return;
}
}
最后:只有二十五分,可能用输入流会减少一点时间,大家可以试一下,顺便给博主说一下。