結果

問題 No.1597 Matrix Sort
ユーザー ecotteaecottea
提出日時 2021-07-29 00:53:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 756 ms / 1,500 ms
コード長 6,633 bytes
コンパイル時間 4,563 ms
コンパイル使用メモリ 232,464 KB
実行使用メモリ 4,800 KB
最終ジャッジ日時 2023-10-11 22:40:41
合計ジャッジ時間 16,101 ms
ジャッジサーバーID
(参考情報)
judge11 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,352 KB
testcase_01 AC 1 ms
4,352 KB
testcase_02 AC 1 ms
4,352 KB
testcase_03 AC 700 ms
4,800 KB
testcase_04 AC 748 ms
4,656 KB
testcase_05 AC 756 ms
4,608 KB
testcase_06 AC 735 ms
4,760 KB
testcase_07 AC 747 ms
4,724 KB
testcase_08 AC 524 ms
4,608 KB
testcase_09 AC 539 ms
4,724 KB
testcase_10 AC 201 ms
4,648 KB
testcase_11 AC 170 ms
4,612 KB
testcase_12 AC 419 ms
4,604 KB
testcase_13 AC 586 ms
4,796 KB
testcase_14 AC 660 ms
4,608 KB
testcase_15 AC 145 ms
4,764 KB
testcase_16 AC 251 ms
4,712 KB
testcase_17 AC 278 ms
4,660 KB
testcase_18 AC 713 ms
4,612 KB
testcase_19 AC 714 ms
4,764 KB
testcase_20 AC 702 ms
4,608 KB
testcase_21 AC 444 ms
4,716 KB
testcase_22 AC 234 ms
4,768 KB
testcase_23 AC 218 ms
4,612 KB
testcase_24 AC 39 ms
4,712 KB
testcase_25 AC 37 ms
4,668 KB
testcase_26 AC 2 ms
4,348 KB
testcase_27 AC 2 ms
4,352 KB
testcase_28 AC 2 ms
4,352 KB
testcase_29 AC 1 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#ifndef HIDDEN_IN_VISUAL_STUDIO // 無意味.折りたたむのが目的.

// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS

// 使えるライブラリの読み込み
#include <bits/stdc++.h>
#include <functional> // function
using namespace std;

// 型名の短縮
using ll = long long;           // -2^63 ~ 2^63 = 9   * 10^18
using ull = unsigned long long; //     0 ~ 2^64 = 1.8 * 10^19
using uint = unsigned int;      //     0 ~ 2^32 = 4   * 10^9
using pii = pair<int, int>;
using pll = pair<ll, ll>;

// 定数の定義
const double PI = 3.141592653589793238462643383279; // 円周率
const vector<int> dx4 = { 1, 0, -1, 0 }; // 4 近傍
const vector<int> dy4 = { 0, 1, 0, -1 };
const vector<int> dx8 = { 1, 1, 0, -1, -1, -1, 0, 1 }; // 8 近傍
const vector<int> dy8 = { 0, 1, 1, 1, 0, -1, -1, -1 };
const ll INFL = (ll)1e18;
const int INF = (int)1e9;
const double EPS = 1e-10; // 許容誤差に応じて調整

// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define rep(i, n) for(int i = 0, i##_len = ((int)(n)); i < i##_len; ++i) // 0 から n-1 まで
#define repi(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = (int)(s), i##_end = ((int)(t)); i >= i##_end; --i) // s から t まで降順
#define repe(i, a) for(const auto& i : (a)) // a の全要素
#define repb(i, d) for(int i = 0, i##_len = (1 << (int)(d)); i < i##_len; ++i) // d ビット全探索
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て
#define Yes(b) if(b){cout << "Yes" << endl;}else{cout << "No" << endl;}
#define Tak(b) if(b){cout << "Takahashi" << endl;}else{cout << "Aoki" << endl;}
#define norm hypot // ノルム(2D, 3D)

// 汎用関数の定義
inline ll pow(ll n, int k) { ll v = 1; rep(i, k) v *= n; return v; } // 工夫が必要なほど k が大きかったらどうせオーバーフローするからこれでいい
inline ll pow(int n, int k) { ll v = 1; rep(i, k) v *= (ll)n; return v; }
template <typename T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す)
template <typename T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す)

// 入出力用の >>, << のオーバーロード
template <typename T, typename U> ostream& operator<< (ostream& os, pair<T, U> p) { os << "(" << p.first << "," << p.second << ")"; return os; } // pair の出力用
template <typename T, typename U> istream& operator>> (istream& is, pair<T, U>& p) { T tmp1; U tmp2; is >> tmp1 >> tmp2; p = { tmp1, tmp2 }; return is; } // pair の入力用

// 手元環境(Visual Studio)
#ifdef _MSC_VER
#define popcount (int)__popcnt // 全ビットにおける 1 の個数
#define popcountll (int)__popcnt64
inline int ctz(uint n) { unsigned long i; _BitScanForward(&i, n); return i; } // 下位ビットに並ぶ 0 の個数
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; } // 最大公約数
#define dump(a) cerr << "[DEBUG] " << endl << a << endl; // デバッグ出力用
#define dump_array(a) cerr << "[DEBUG]" << endl; for (auto x : a) {cout << x << " ";} cout << endl;
#define dump_array2(a) cerr << "[DEBUG]" << endl; for (auto c : a) {for (auto x : c) {cout << x << " ";} cout << endl;}
// 提出用(GCC)
#else
#define popcount (int)__builtin_popcount
#define popcountll (int)__builtin_popcountll
#define ctz __builtin_ctz
#define gcd __gcd
#define dump(a) 
#define dump_array(a) 
#define dump_array2(a) 
#endif

#endif // 無意味.折りたたむのが目的.


// AtCoder 専用
#include <atcoder/all>
using namespace atcoder;

ostream& operator<< (ostream& os, modint1000000007 x) { os << x.val(); return os; } // mint の出力用
ostream& operator<< (ostream& os, modint998244353 x) { os << x.val(); return os; }
ostream& operator<< (ostream& os, modint x) { os << x.val(); return os; }
istream& operator>> (istream& is, modint1000000007& x) { ll tmp; is >> tmp; x = tmp; return is; } // mint の入力用
istream& operator>> (istream& is, modint998244353& x) { ll tmp; is >> tmp; x = tmp; return is; }
istream& operator>> (istream& is, modint& x) { ll tmp; is >> tmp; x = tmp; return is; }

// mint で使いたい法によってここを切り替える
using mint = modint1000000007;
//using mint = modint998244353;
//using mint = modint; // modint::set_mod(10000); // mint の法の指定


//【めぐる式二分探索】O(log |ok - ng|)
/*
* 条件 okQ() を満たす要素 ok と満たさない要素 ng との境界を二分探索する.
* 境界に隣り合うような条件を満たす要素(ok 側)の位置を返す.
*/
template <typename T> // T は int, ll など
T binary_search(T ok, T ng, function<bool(T)>& okQ) {
	// 境界が決定するまで
	while (abs(ok - ng) > 1) {
		// 区間の中間
		T mid = (ok + ng) / 2;

		// 中間が OK かどうかに応じて区間を縮小する.
		if (okQ(mid)) {
			ok = mid;
		}
		else {
			ng = mid;
		}
	}
	return ok;
}

int main() {
	int n;
	ll k, p;
	cin >> n >> k >> p;

	vector<ll> a(n);
	rep(i, n) {
		cin >> a[i];
	}
	sort(all(a));
	dump_array(a);

	vector<ll> b(n);
	rep(i, n) {
		cin >> b[i];
	}
	dump_array(b);

	// k 番目の数が v 以上なら true,さもなくば false.
	// 言い換えれば,v 未満の数が k 個未満なら true, さもなくば false.
	function<bool(ll)> okQ = [&](ll v) {
		ll cnt = 0;

		// 各 b[i] について,(a + b[i]) mod p < v なる A の要素 a を数える.
		// これを同値変形していくと
		//		0 <= a + b[i] < v || p <= a + b[i] < p + v
		//		⇔ a < v - b[i] || p - b[i] <= a < p + v - b[i]
		// となる.
		rep(i, n) {
			auto it1 = lower_bound(all(a), v - b[i]);
			auto it2 = lower_bound(all(a), p - b[i]);
			auto it3 = lower_bound(all(a), p + v - b[i]);

			cnt += (ll)distance(a.begin(), it1);
			cnt += (ll)distance(it2, it3);
		}

		return cnt < k;
	};

	auto res = binary_search(0LL, p, okQ);

	cout << res << endl;
}

/* TLE 畳み込み
int main() {
	int n, k, p;
	cin >> n >> k >> p;

	vector<ll> a_cnt(p), b_cnt(p);
	rep(i, n) {
		int a;
		cin >> a;

		a_cnt[a % p]++;
	}
	rep(i, n) {
		int b;
		cin >> b;

		b_cnt[b % p]++;
	}

	auto m_cnt = convolution_ll(a_cnt, b_cnt);

	ll cnt_sum = 0;
	rep(i, p - 1) {
		cnt_sum += m_cnt[i] + m_cnt[i + p];
		if (k <= cnt_sum) {
			cout << i << endl;
			return 0;
		}
	}
	cout << p - 1 << endl;
}
*/
0