【刷题记录】 Codeforces Round #181 (Div. 2) Coach

目录

一. 问题描述

二. 题解代码

---

一. 问题描述

Problem Description

教练要训练n个队员 ， 要把这些队员分组为三人一组 ，题目保证n一定是三的倍数 ， 但是有些队员 i 要求跟队员 j 一队，i和j就必须一队。有m条队员要求 ， 要求输出可能的一组分队情况。 如果不可能则输出-1 。 3<=n<=48 。

二. 题解代码

        我们可以先用并查集把想要一队的连接到一个网络里，然后以圈子根为编号把所有该网络人加入到一个数组里，然后分析这个数组，如果有数组人员数>3则说明无解。把所有自由元都放到一个数组里，在不足三人的圈子里加入自由元补足人数，最后输出即可。比较好想的思路，但处理上有点麻烦，我用的双端队列。

#include <iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<deque>
using namespace std;
typedef long long LL;
const int maxn = 100 + 10;
deque<int> num[maxn];
int pre[maxn];
int finded(int a)
{
    int v = a;
    while(v!=pre[v]){
        v = pre[v];
    }
    int j = a;
    while(j!=v){
        int temp = pre[j];
        pre[j] = v;
        j = temp;
    }
    return v;
}
void join(int a,int b)
{
    int fx = finded(a);
    int fy = finded(b);
    if(fx!=fy){
        pre[fx] = fy;
    }
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 1;i<=n;i++)pre[i] = i;
    for(int j = 0;j<m;j++){
        int p,q;
        scanf("%d%d",&p,&q);
        join(p,q);
    }
    for(int i = 1;i<=n;i++){
        num[finded(i)].push_back(i);
    }
    bool judge = false;
    for(int j = 1;j<=n;j++){
        if(num[j].size()==1){num[0].push_back(j);num[j].pop_back();}
        else if(num[j].size()>3){judge = true;break;}
    }
    if(judge)printf("-1\n");
    else{
        for(int k = 1;k<=n;k++){
            if(num[k].size()==0||num[k].size()==3)continue;
            else if(num[k].size()<3){
                while(num[k].size()<3){
                    if(!num[0].empty()){
                    int t = num[0].front();
                    num[0].pop_front();
                    num[k].push_back(t);}
                    else {printf("-1\n");return 0;}
                }
            }
        }
        if(num[0].size()>0){
            int m = 0;
            for(int i = 0;i<num[0].size();i++){
                m++;
                if(m==1)printf("%d",num[0][i]);
                else
                 printf(" %d",num[0][i]);
                if(m==3){printf("\n");m = 0;}
            }
        }
        for(int j = 1;j<=n;j++){
            if(num[j].size()==3){
                for(int i = 0;i<3;i++){
                    if(i==0)printf("%d",num[j][i]);
                    else printf(" %d",num[j][i]);
                }
                printf("\n");
            }
        }
    }
    return 0;
}