結果

問題 No.697 池の数はいくつか
ユーザー wunderkammer2wunderkammer2
提出日時 2020-02-26 21:56:44
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 2,662 bytes
コンパイル時間 1,083 ms
コンパイル使用メモリ 104,768 KB
実行使用メモリ 657,612 KB
最終ジャッジ日時 2024-11-25 15:55:37
合計ジャッジ時間 40,413 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 MLE -
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 1 ms
6,816 KB
testcase_06 AC 1 ms
6,820 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 2 ms
6,816 KB
testcase_09 AC 2 ms
6,816 KB
testcase_10 AC 2 ms
6,820 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 2 ms
6,820 KB
testcase_13 AC 2 ms
6,820 KB
testcase_14 AC 2 ms
6,820 KB
testcase_15 AC 2 ms
6,820 KB
testcase_16 AC 2 ms
6,820 KB
testcase_17 AC 2 ms
6,816 KB
testcase_18 AC 2 ms
6,820 KB
testcase_19 AC 2 ms
6,816 KB
testcase_20 AC 2 ms
6,820 KB
testcase_21 AC 1 ms
6,816 KB
testcase_22 AC 2 ms
6,816 KB
testcase_23 AC 2 ms
6,816 KB
testcase_24 AC 490 ms
34,560 KB
testcase_25 AC 505 ms
34,504 KB
testcase_26 AC 499 ms
34,600 KB
testcase_27 AC 520 ms
34,428 KB
testcase_28 AC 516 ms
34,584 KB
testcase_29 TLE -
testcase_30 AC 5,167 ms
285,968 KB
testcase_31 TLE -
testcase_32 AC 5,240 ms
286,764 KB
testcase_33 AC 5,300 ms
286,928 KB
testcase_34 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<algorithm>
#include<cmath>
#include<cstdio>
#include<functional>
#include<iomanip>
#include<iostream>
#include<map>
#include<numeric>
#include<queue>
#include<set>
#include<string>
#include<utility>
#include<vector>

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll MOD = 1000000007;
#define rep(i,n) for(int i=0;i<n;i++)
#define repl(i,s,e) for(int i=s;i<e;i++)
#define reple(i,s,e) for(int i=s;i<=e;i++)
#define revrep(i,n) for(int i=n-1;i>=0;i--)
#define all(x) (x).begin(),(x).end()

int H, W;


class UnionFind
{
	//par[i]：ノードiの親
	//count[i]：iが親の時、iが属するグループのノード数
	vector<int> par;
	vector<int> count;

	int m_groupCount = 0;

public:
	UnionFind(int N)
	{
		par.push_back(0);
		count.push_back(0);

		m_groupCount = N;

		for (int i = 1; i <= N; i++)
		{
			//最初は全てが根であるとして初期化
			par.push_back(i);
			count.push_back(1);
		}
	}

	//データxが属する木の根を再帰で得る：root(x) = {xの木の根}
	int root(int x)
	{
		if (par[x] == x) return x;

		//ついでに親を張り替え
		return par[x] = root(par[x]);
	}

	//xとyの木を併合
	void unite(int x, int y)
	{
		int rx = root(x);
		int ry = root(y);
		if (rx == ry) return;
		//xとyの根が同じでない(=同じ木にない)時：xの根rxをyの根ryにつける
		par[rx] = ry;

		//要素数の変更
		count[ry] += count[rx];
		count[rx] = 0;

		//木の併合によってグループが1つ減る
		m_groupCount--;
	}

	// 2つのデータx, yが属する木が同じならtrueを返す
	bool same(int x, int y)
	{
		return root(x) == root(y);
	}

	//指定したデータxが属するグループの要素数を返す
	int size(int x)
	{
		return count[root(x)];
	}

	//UnionFindのグループの個数を返す
	int groupCount()
	{
		return m_groupCount;
	}
};

bool IsInRange(int w, int h)
{
	return 0 <= w && w <= W - 1 && 0 <= h && h <= H - 1;
}

int ToIndex(int w, int h)
{
	return h * W + w;
}

int main()
{
	cin >> H >> W;

	UnionFind uf(H * W);

	set<int> s;

	rep(h, H)
	{
		rep(w, W)
		{
			int a;
			cin >> a;

			if (a != 1) continue;

			int idx = ToIndex(w, h);

			//1の場合setで保存
			s.insert(idx);

			//上と左だけ結合判定
			if (IsInRange(w - 1, h))
			{
				int i = ToIndex(w - 1, h);
				if(s.count(i) > 0) uf.unite(idx, i);
			}

			if (IsInRange(w, h - 1))
			{
				int i = ToIndex(w, h - 1);
				if (s.count(i) > 0) uf.unite(idx, i);
			}
		}
	}
	

	//個数確認
	//親の個数をカウント
	ll count = 0;

	for (auto x : s)
	{
		if (uf.root(x) == x) count++;
	}
	
	cout << count << endl;

	return 0;
}
0