結果

問題 No.697 池の数はいくつか
ユーザー wunderkammer2wunderkammer2
提出日時 2020-02-26 22:05:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,209 ms / 6,000 ms
コード長 2,751 bytes
コンパイル時間 1,551 ms
コンパイル使用メモリ 100,940 KB
実行使用メモリ 137,192 KB
最終ジャッジ日時 2024-11-08 08:36:04
合計ジャッジ時間 17,696 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 2 ms
5,248 KB
testcase_14 AC 2 ms
5,248 KB
testcase_15 AC 3 ms
5,248 KB
testcase_16 AC 2 ms
5,248 KB
testcase_17 AC 2 ms
5,248 KB
testcase_18 AC 2 ms
5,248 KB
testcase_19 AC 2 ms
5,248 KB
testcase_20 AC 2 ms
5,248 KB
testcase_21 AC 2 ms
5,248 KB
testcase_22 AC 3 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 211 ms
15,252 KB
testcase_25 AC 212 ms
15,376 KB
testcase_26 AC 214 ms
15,252 KB
testcase_27 AC 211 ms
15,220 KB
testcase_28 AC 210 ms
15,128 KB
testcase_29 AC 2,036 ms
137,192 KB
testcase_30 AC 2,197 ms
137,064 KB
testcase_31 AC 2,029 ms
137,064 KB
testcase_32 AC 2,209 ms
137,060 KB
testcase_33 AC 2,187 ms
137,192 KB
testcase_34 AC 2,185 ms
137,064 KB
権限があれば一括ダウンロードができます

ソースコード

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;

	vector<vector<int>> A(W, vector<int>(H, 0));

	rep(h, H)
		rep(w, W)
		cin >> A[w][h];

	UnionFind uf(H * W);

	rep(h, H)
	{
		rep(w, W)
		{
			if (A[w][h] != 1) continue;

			int idx = ToIndex(w, h);

			//上と左だけ結合判定
			if (IsInRange(w - 1, h) && A[w - 1][h] == 1)
			{
				int i = ToIndex(w - 1, h);
				uf.unite(idx, i);
			}

			if (IsInRange(w, h - 1) && A[w][h - 1] == 1)
			{
				int i = ToIndex(w, h - 1);
				uf.unite(idx, i);
			}
		}
	}
	

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

	rep(h, H)
	{
		rep(w, W)
		{
			if (A[w][h] != 1) continue;

			int idx = ToIndex(w, h);
			if (uf.root(idx) == idx) count++;
		}
	}
	
	cout << count << endl;

	return 0;
}
0