2D-Array( $2$ 次元配列) (structure/2d_array.cpp)

• View this file on GitHub
• View document part on GitHub
• Last update: 2021-02-26 21:39:59+09:00

なにこれ

入出力・宣言を簡潔に行える $2$ 次元配列．

コンストラクタ

• matrix()：$0$ 行 $0$ 列の $2$ 次元配列を作る．
• matrix(h, w)：h 行 w 列の $2$ 次元配列を作る．
• matrix(h, w, init)：init を初期値とする h 行 w 列の $2$ 次元配列を作る．
• matrix(m_init)：m_initを初期値とする $2$ 次元配列を作る．

メンバ関数

• assign(h, w)：h 行 w 列の $2$ 次元配列を作る．
• assign(h, w, init)：init を初期値とする h 行 w 列の $2$ 次元配列を作る．
• size()：配列の縦のサイズを返す．
• in()：現在の要素サイズ分だけ標準入力する．
• in(h, w)：h 行 w 列の $2$ 次元配列を作り標準入力する．
• out()：要素を空白区切りで標準出力する．
• operator[](idx)：idx 番目の要素である $1$ 次元配列の参照を返す．vector を用いた多次元配列と同様に扱える．
• rangeout(x ,y)：x 行 y 列目の要素が存在しない場合は true ，存在する場合は false を返す．

計算量

$H$ 行 $W$ 列の行列を扱うとする．

• コンストラクタ：$O(HW)$
• assign()：$O(HW)$
• size()：$O(1)$
• in()：$O(HW)$
• out()：$O(HW)$
• operator[](idx)：$O(1)$
• rangeout(x, y)：$O(1)$

参考

• 初期化子リスト - cpprefjp C++日本語リファレンス

Required by

• 2D-Cumulative-Sum( $2$ 次元累積和) (dp/2d_cumulative_sum.cpp)
• Doubling(ダブリング) (dp/doubling.cpp)
• Centroid Decomposition (木の重心分解) (graph/centroid_decomposition.cpp)
• Dijkstra(経路復元付ダイクストラ) (graph/dijkstra.cpp)
• Topological-Sort(トポロジカルソート) (graph/topological_sort.cpp)

Verified with

• test/aoj_0116.cpp
• test/aoj_0341.cpp
• test/aoj_0560.cpp
• test/aoj_2429.cpp
• test/aoj_GRL_1_A.cpp
• test/aoj_GRL_4_B.cpp
• test/aoj_ITP1_6_B.cpp
• test/atc_abc167_d.cpp
• test/atc_abc175_d.cpp
• test/atc_abc291_ex.cpp
• test/atc_nikkei2019-qual_d.cpp
• test/cf_321_c.cpp
• test/yosupo_shortest_path.cpp
• test/yuki_0177.cpp
• test/yuki_0497.cpp
• test/yuki_1013.cpp
• test/yuki_1097.cpp

Code

#pragma once

#ifndef call_include
#define call_include
#include <bits/stdc++.h>
using namespace std;
#endif

template<typename T> struct v2d {
private:
	vector<vector<T>> m;

public:
	v2d() {}
	v2d(int h, int w) : m(h, vector<T>(w)) {}
	v2d(int h, int w, const T &init) : m(h, vector<T>(w, init)) {}
	v2d(const initializer_list<initializer_list<T>> m_init) : m(m_init.begin(), m_init.end()) {}

	void assign(int h, int w) {
		m.assign(h, vector<T>(w));
	}
	void assign(int h, int w, const T init) {
		m.assign(h, vector<T>(w, init));
	}

	inline int size() const {
		return m.size();
	}

	void in() {
		for(vector<T> &v : m)
			for(T &val : v) cin >> val;
	}
	void in(int h, int w) {
		m.resize(h, vector<T>(w));
		in();
	}

	void out() {
		int h = m.size();
		for(vector<T> &v : m) {
			int sz = v.size();
			for(int j = 0; j < sz; j++) {
				cout << v[j] << (j == sz - 1 ? '\n' : ' ');
			}
		}
		cout << flush;
	}

	inline vector<T> &operator[](int idx) {
		assert(0 <= idx && idx < m.size());
		return m[idx];
	}

	bool rangeout(int x, int y) {
		if(x < 0 || y < 0 || x >= size() || y >= m[x].size()) return true;
		return false;
	}
};

#line 2 "structure/2d_array.cpp"

#ifndef call_include
#define call_include
#include <bits/stdc++.h>
using namespace std;
#endif

template<typename T> struct v2d {
private:
	vector<vector<T>> m;

public:
	v2d() {}
	v2d(int h, int w) : m(h, vector<T>(w)) {}
	v2d(int h, int w, const T &init) : m(h, vector<T>(w, init)) {}
	v2d(const initializer_list<initializer_list<T>> m_init) : m(m_init.begin(), m_init.end()) {}

	void assign(int h, int w) {
		m.assign(h, vector<T>(w));
	}
	void assign(int h, int w, const T init) {
		m.assign(h, vector<T>(w, init));
	}

	inline int size() const {
		return m.size();
	}

	void in() {
		for(vector<T> &v : m)
			for(T &val : v) cin >> val;
	}
	void in(int h, int w) {
		m.resize(h, vector<T>(w));
		in();
	}

	void out() {
		int h = m.size();
		for(vector<T> &v : m) {
			int sz = v.size();
			for(int j = 0; j < sz; j++) {
				cout << v[j] << (j == sz - 1 ? '\n' : ' ');
			}
		}
		cout << flush;
	}

	inline vector<T> &operator[](int idx) {
		assert(0 <= idx && idx < m.size());
		return m[idx];
	}

	bool rangeout(int x, int y) {
		if(x < 0 || y < 0 || x >= size() || y >= m[x].size()) return true;
		return false;
	}
};

Back to top page