글 읽기 - 91%에서 계속 걸려요 ......ㅠㅠ도와주세요 ㅜ(유니온파인드)

blackpoplar1004 3년 전 0

91%에서 계속 걸려요 ......

#include<iostream>
#include<vector>
using namespace std;
int aParent[1'000'001];

int getParent(int x)
{
	if (aParent[x] == x) return x;
	else return aParent[x] = getParent(aParent[x]);
}

void UnionFind(int _a, int _b) //이거 때문에 틀렸음... ? 
{
	_a = getParent(_a);
	_b = getParent(_b);

	if (_a < _b)aParent[_b] = _a;
	else aParent[_a] = _b;
}

int TrueAndFalse(int _a, int _b)
{
	int a = getParent(_a);
	int b = getParent(_b);

	if (a == b) return 1;
	else return 0;
}
int main(void)
{
	ios_base::sync_with_stdio(false);
	cin.tie(0);

	int n, m;
	cin >> n >> m;

	int control, data;
	vector<int> vec;

	for (int i = 0; i <= m; i++)
	{
		aParent[i] = i;
	}

	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
		{
			cin >> data;
			if (data)
			{
				UnionFind(i, j);
			}
		}

	vector<bool> bec;

	for (int i = 0; i < m; i++)
	{
		bool check = false;

		cin >> data;

		if (i != 0)
		{
			for (int j = 1; j <= n; j++)
			{
				if (TrueAndFalse(j, data) && j != data)
				{
					check = true;
				}
			}
		}
		bec.push_back(check);
	}

	bool final = true;
	for (int i = 1; i < m; i++)
	{
		if (!bec[i]) final = false;
	}


	if (final)
	{
		cout << "YES" << "\n";
	}
	else
	{
		cout << "NO" << "\n";
	}
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

91%에서 계속 걸려요 ......ㅠㅠ도와주세요 ㅜ(유니온파인드)

blackpoplar1004 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말