#211552 (C++14) No.348 カゴメカゴメ

提出ソース

結果

問題	No.348 カゴメカゴメ
ユーザー	FF256grhy
提出日時	2017-10-26 07:00:08
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	454 ms / 2,000 ms
コード長	3,888 bytes
コンパイル時間	1,974 ms
コンパイル使用メモリ	182,204 KB
実行使用メモリ	103,020 KB
最終ジャッジ日時	2024-05-01 14:50:14
合計ジャッジ時間	7,919 ms
ジャッジサーバーID （参考情報）	judge5 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	199 ms 54,004 KB
testcase_01	AC	11 ms 27,152 KB
testcase_02	AC	359 ms 71,688 KB
testcase_03	AC	454 ms 103,020 KB
testcase_04	AC	11 ms 27,028 KB
testcase_05	AC	10 ms 27,160 KB
testcase_06	AC	10 ms 27,028 KB
testcase_07	AC	10 ms 27,188 KB
testcase_08	AC	10 ms 27,256 KB
testcase_09	AC	10 ms 27,136 KB
testcase_10	AC	9 ms 27,188 KB
testcase_11	AC	10 ms 27,244 KB
testcase_12	AC	13 ms 27,524 KB
testcase_13	AC	12 ms 27,360 KB
testcase_14	AC	10 ms 27,176 KB
testcase_15	AC	76 ms 36,804 KB
testcase_16	AC	10 ms 27,108 KB
testcase_17	AC	10 ms 27,096 KB
testcase_18	AC	10 ms 26,980 KB
testcase_19	AC	10 ms 27,116 KB
testcase_20	AC	11 ms 27,200 KB
testcase_21	AC	10 ms 27,224 KB
testcase_22	AC	9 ms 27,044 KB
testcase_23	AC	260 ms 71,100 KB
testcase_24	AC	256 ms 71,324 KB
testcase_25	AC	258 ms 70,656 KB
testcase_26	AC	259 ms 70,216 KB
testcase_27	AC	255 ms 70,072 KB
testcase_28	AC	264 ms 70,060 KB
testcase_29	AC	257 ms 69,800 KB
testcase_30	AC	267 ms 69,808 KB
testcase_31	AC	256 ms 71,568 KB
testcase_32	AC	253 ms 70,216 KB
testcase_33	AC	30 ms 29,992 KB
testcase_34	AC	30 ms 29,888 KB
testcase_35	AC	31 ms 29,876 KB
testcase_36	AC	32 ms 29,864 KB
testcase_37	AC	32 ms 29,868 KB
testcase_38	AC	31 ms 29,868 KB
testcase_39	AC	29 ms 29,868 KB
testcase_40	AC	30 ms 29,992 KB
testcase_41	AC	30 ms 29,864 KB
testcase_42	AC	30 ms 29,992 KB
testcase_43	AC	14 ms 27,472 KB
testcase_44	AC	15 ms 27,372 KB
testcase_45	AC	14 ms 27,372 KB
testcase_46	AC	14 ms 27,640 KB
testcase_47	AC	12 ms 27,424 KB
testcase_48	AC	12 ms 27,240 KB
testcase_49	AC	14 ms 27,496 KB
testcase_50	AC	14 ms 27,696 KB
testcase_51	AC	13 ms 27,496 KB
testcase_52	AC	11 ms 27,372 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;

typedef long long   signed int LL;
typedef long long unsigned int LU;

#define incID(i, l, r) for(int i = (l)    ; i <  (r); i++)
#define incII(i, l, r) for(int i = (l)    ; i <= (r); i++)
#define decID(i, l, r) for(int i = (r) - 1; i >= (l); i--)
#define decII(i, l, r) for(int i = (r)    ; i >= (l); i--)
#define  inc(i, n) incID(i, 0, n)
#define inc1(i, n) incII(i, 1, n)
#define  dec(i, n) decID(i, 0, n)
#define dec1(i, n) decII(i, 1, n)

#define inII(v, l, r) ((l) <= (v) && (v) <= (r))
#define inID(v, l, r) ((l) <= (v) && (v) <  (r))

#define PB push_back
#define EB emplace_back
#define MP make_pair
#define FI first
#define SE second
#define PQ priority_queue

#define  ALL(v)  v.begin(),  v.end()
#define RALL(v) v.rbegin(), v.rend()
#define  FOR(it, v) for(auto it =  v.begin(); it !=  v.end(); ++it)
#define RFOR(it, v) for(auto it = v.rbegin(); it != v.rend(); ++it)

template<typename T> bool   setmin(T & a, T b) { if(b <  a) { a = b; return true; } else { return false; } }
template<typename T> bool   setmax(T & a, T b) { if(b >  a) { a = b; return true; } else { return false; } }
template<typename T> bool setmineq(T & a, T b) { if(b <= a) { a = b; return true; } else { return false; } }
template<typename T> bool setmaxeq(T & a, T b) { if(b >= a) { a = b; return true; } else { return false; } }
template<typename T> T gcd(T a, T b) { return (b == 0 ? a : gcd(b, a % b)); }
template<typename T> T lcm(T a, T b) { return a / gcd(a, b) * b; }

// ---- ----

class UnionFind {
public:
	void init(int size);
	int unite(int v, int w);
	int find(int v);
	int get_size_eq(int v);
	int get_size_part();
	virtual ~UnionFind();

private:
	int *parent;
	int *size_eq;
	int size_part;
};

void UnionFind::init(int size) {
	parent = new int[size];
	size_eq = new int[size];
	for(int i = 0; i < size; i++) { parent[i] = -1; size_eq[i] = 1; }
	size_part = size;
}

int UnionFind::unite(int v, int w) {
	int fv = find(v);
	int fw = find(w);
	if(fv == fw) { return fv; }
	
	size_part--;
	int &sv = size_eq[fv];
	int &sw = size_eq[fw];
	
	if(sv < sw) {
		parent[fv] = fw;
		sw += sv;
		return fw;
	} else {
		parent[fw] = fv;
		sv += sw;
		return fv;
	}
}

int UnionFind::find(int v) {
	if(parent[v] == -1) { return v; }
	return parent[v] = find( parent[v] );
}

int UnionFind::get_size_eq(int v) {
	return size_eq[ find(v) ];
}

int UnionFind::get_size_part() {
	return size_part;
}

UnionFind::~UnionFind() {
	delete[] parent;
	delete[] size_eq;
}

// ----

#define UQ(v) v.erase(unique(ALL(v)), v.end())

int n, m;
string s[1003];
UnionFind uf;
vector<int> v[1003 * 1003];

pair<int, int> dfs(int pos, int par, int t) {
	auto p1 = MP(0, 0);
	inc(i, v[pos].size()) {
		if(v[pos][i] == par) { continue; }
		auto p2 = dfs(v[pos][i], pos, 1 - t);
		p1.FI += p2.FI;
		p1.SE += p2.SE;
	}
	if(t == 0) { return p1; }
	swap(p1.FI, p1.SE);
	p1.FI += uf.get_size_eq(pos);
	setmax(p1.FI, p1.SE);
	return p1;
}

int main() {
	cin >> n >> m;
	s[0] = s[1] = s[n + 2] = string(m + 3, '.');
	incID(i, 2, n + 2) { cin >> s[i]; s[i] = ".." + s[i] + "."; }
	s[0][0] = 'x';
	
	n += 3;
	m += 3;
	uf.init(n * m);
	vector<pair<int, int>> vp;
	int di[4] = { 0,  1, 1, 1 };
	int dj[4] = { 1, -1, 0, 1 };
	inc(q, 2) {
	inc(i, n) {
	inc(j, m) {
		inc(k, 4) {
			int ii = i + di[k]; if(! inID(ii, 0, n)) { continue; }
			int jj = j + dj[k]; if(! inID(jj, 0, m)) { continue; }
			if(q == 0) {
				if(
					(s[i][j] == 'x' && s[ii][jj] == 'x') ||
					(s[i][j] == '.' && s[ii][jj] == '.' && k % 2 == 0)
				) { uf.unite(m * i + j, m * ii + jj); }
			} else {
				if(s[i][j] != s[ii][jj]) {
					vp.EB(uf.find(m * i + j), uf.find(m * ii + jj));
					vp.EB(uf.find(m * ii + jj), uf.find(m * i + j));
				}
			}
		}
	}
	}
	}
	sort(ALL(vp));
	UQ(vp);
	inc(i, vp.size()) { v[vp[i].FI].PB(vp[i].SE); }
	
	cout << dfs(uf.find(m * 0 + 0), -1, 1).SE << endl;
	
	return 0;
}

yukicoder

結果

テストケース

ソースコード