排座位 (25分)

    科技2022-07-17  112

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

    输入格式: 输入第一行给出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

    #include <bits/stdc++.h> using namespace std; int f[110]; int e[110][110]; int getf(int v) { if(f[v]==v) return v; f[v]=getf(f[v]); return f[v]; } void marge(int a,int b) { int t1,t2; t1=getf(a); t2=getf(b); if(t1!=t2) f[t2]=t1; } int main() { int n,m,k,i,u,v,w; /* 使用无向图e来直接储存u和v的关系 使用并查集f来储存“朋友的朋友” */ memset(e,0,sizeof(0)); scanf("%d %d %d",&n,&m,&k); for(i=1; i<=n; i++) f[i]=i; for(i=1; i<=m; i++) { scanf("%d %d %d",&u,&v,&w); e[u][v]=e[v][u]=w; if(w==1) marge(u,v); } for(i=1; i<=k; i++) { scanf("%d %d",&u,&v); if((getf(u)==getf(v)&&e[u][v]!=-1)||e[u][v]==1) printf("No problem\n"); else if(getf(u)!=getf(v)&&e[u][v]==0) printf("OK\n"); else if(getf(u)==getf(v)&&e[u][v]==-1) printf("OK but...\n"); else if(getf(u)!=getf(v)&&e[u][v]==-1) printf("No way\n"); } return 0; }
    Processed: 0.009, SQL: 8