D - Neighbors（并查集

题意：

n个人排队，m个意愿，每个意愿给两个数a,b，意思是a和b想站一起，问最后能否完全符合他们的意愿排成一队。

思路：

• 当一个人想站一起的人数超过两人那么肯定是no
• 如果关系构成了一个环，那么肯定也是no，当一个人想和两个人站一起但那两个人也想站一起的情况。
  之前自己也想到了，用了建图＋拓扑排序+并查集，想复杂了。

错误代码

#include<bits/stdc++.h> 
#define int long long
using namespace std;
const int N = 1e5 + 10, M = N * 2;
int h[N * 2], e[M], ne[M], idx;
int q[M * 2];
int n, m;


int p[N]; //存储每个点的祖宗节点

// 返回x的祖宗节点
int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

// 初始化，假定节点编号是1~n


// 合并a和b所在的两个集合：



void add(int a, int b)  // 添加一条边a->b
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
int d[M * 2];
int dd[M * 2];
bool topsort()
{
    int hh = 0, tt = -1;

    // d[i] 存储点i的入度
    for (int i = 0; i <= n; i ++ )
    {
            if (!d[i]) q[ ++ tt] = i;
            
        }
        
    while (hh <= tt)
    {
        int t = q[hh ++ ];

        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (-- d[j] == 0)
                q[ ++ tt] = j;
        }
    }

    // 如果所有点都入队了，说明存在拓扑序列；否则不存在拓扑序列。
    return tt == n;
}

signed main()
{
        memset(h, -1, sizeof h);
        cin >> n >> m;
        for (int i = 1; i <= n; i ++ ) p[i] = i;
        for (int i = 1; i <= m; i ++ )
        {
                int a, b;
                cin >> a >> b;
                add(a, b);
                d[b] ++;
                dd[a] ++;
                dd[b] ++;
                p[find(a)] = find(b);
        }
        bool f = 0;
        int cnt = 0;
        for (int i = 1; i <= n; i ++ )
        {
                if (dd[i] == 2) cnt ++;
                if (dd[i] > 2 && !f)
                {
                        f = 1;
                        cout << "No";
                        return 0;
                }
        }
        if (cnt > n - 2)
        {
                cout << "No"; 
                return 0;
        }
        set<int> qq;
        for (int i = 1; i <= n; i ++ )
        {
                qq.insert(find(i));
        }
        for(auto i : qq)
        {
//                cout << i << " ";
                add(0, i);
                d[i] ++;
        }
        if (topsort()) cout << "Yes" << endl;
        else cout << "No" << endl;
    
        return 0;
}

正确代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int d[N];
int f[N];
int p[N]; //存储每个点的祖宗节点

// 返回x的祖宗节点
int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}




int main()
{
	int n, m;
	cin >> n >> m;
	bool f = 0;
	// 初始化，假定节点编号是1~n
	for (int i = 1; i <= n; i ++ ) p[i] = i;
	for (int i = 1; i <= m; i ++ )
	{
		int a, b; cin >> a >> b;
		// 合并a和b所在的两个集合：
		if (find(a) == find(b))
		{
			cout << "No" << endl;
			return 0;
		}
		p[find(a)] = find(b);
		d[a] ++;
		d[b] ++;
	}
	for (int i = 1; i <= n; i ++ )
	{
		if (d[i] > 2) 
		{
			cout << "No" << endl;
			return 0;
		}
	}
	cout << "Yes" << endl;
	return 0;
}