結果

問題 No.186 中華風 (Easy)
ユーザー SHIJOUSHIJOU
提出日時 2020-10-10 20:34:58
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 3,372 bytes
コンパイル時間 2,129 ms
コンパイル使用メモリ 202,380 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-09-27 22:38:27
合計ジャッジ時間 2,719 ms
ジャッジサーバーID
(参考情報)
judge14 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 1 ms
4,376 KB
testcase_11 AC 1 ms
4,380 KB
testcase_12 AC 1 ms
4,376 KB
testcase_13 AC 1 ms
4,380 KB
testcase_14 AC 1 ms
4,380 KB
testcase_15 AC 2 ms
4,380 KB
testcase_16 AC 2 ms
4,380 KB
testcase_17 AC 1 ms
4,376 KB
testcase_18 AC 2 ms
4,380 KB
testcase_19 AC 1 ms
4,376 KB
testcase_20 AC 1 ms
4,376 KB
testcase_21 AC 1 ms
4,380 KB
testcase_22 AC 2 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

//#define _GLIBCXX_DEBUG
#include <bits/stdc++.h>
#define rep(i, n) for(int i=0; i<n; ++i)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
using namespace std;
using ll = int64_t;
using ld = long double;
using P = pair<int, int>;
using vs = vector<string>;
using vi = vector<int>;
using vvi = vector<vi>;
template<class T> using PQ = priority_queue<T>;
template<class T> using PQG = priority_queue<T, vector<T>, greater<T> >;
const int INF = 0xccccccc;
const ll LINF = 0xcccccccccccccccLL;
template<typename T1, typename T2>
inline bool chmax(T1 &a, T2 b) {return a < b && (a = b, true);}
template<typename T1, typename T2>
inline bool chmin(T1 &a, T2 b) {return a > b && (a = b, true);}
template<typename T1, typename T2>
istream &operator>>(istream &is, pair<T1, T2> &p) { return is >> p.first >> p.second;}
template<typename T1, typename T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p) { return os << p.first << ' ' << p.second;}

struct barrett {
	unsigned int _m;
	uint64_t im;
	barrett(unsigned int m):_m(m), im(uint64_t(-1)/m + 1) {}
	unsigned int umod() const {return _m;}
	unsigned int mul(unsigned int a, unsigned int b) const {
		uint64_t z = a;
		z *= b;
		uint64_t x = uint64_t(((unsigned __int128)(z)*im) >> 64);
		unsigned int v = (unsigned int)(z-x*_m);
		if(_m <= v) v += _m;
		return v;
	}
};

constexpr int64_t safe_mod(int64_t x, int64_t m) {
	x %= m;
	if(x < 0) x += m;
	return x;
}

constexpr pair<int64_t, int64_t> inv_gcd(int64_t a, int64_t b) {
	a = safe_mod(a, b);
	if(a == 0) return {b, 0};
	int64_t s = b, t = a;
	int64_t m0 = 0, m1 = 1;
	while(t) {
		int64_t u = s/t;
		s -= t*u;
		m0 -= m1*u;
		int64_t tmp = s;
		s = t;
		t = tmp;
		tmp = m0;
		m0 = m1;
		m1 = tmp;
	}
	if(m0 < 0) m0 += b/s;
	return {s, m0};
}

int64_t pow_mod(int64_t x, int64_t n, int m) {
	if(m == 1) return 0;
	barrett bt((unsigned int)(m));
	unsigned int r = 1, y = (unsigned int)(safe_mod(x, m));
	while(n) {
		if(n&1) r = bt.mul(r, y);
		y = bt.mul(y, y);
		n >>= 1;
	}
	return r;
}

int64_t inv_mod(int64_t x, int64_t m) {
	pair<int64_t, int64_t> z = inv_gcd(x, m);
	assert(z.first == 1);
	return z.second;
}

//(y, z) answer mod z = y, 無理な時は(-1, -1)
pair<int64_t, int64_t> crt(const vector<int64_t> &r, const vector<int64_t> &m) {
	int n = int(r.size());
	int64_t r0 = 0, m0 = 1;
	for(int i = 0; i < n; i++) {
		assert(1 <= m[i]);
		int64_t r1 = safe_mod(r[i], m[i]), m1 = m[i];
		if(m0 < m1) {
			swap(r0, r1);
			swap(m0, m1);
		}
		if(m0 % m1 == 0) {
			if(r0 % m1 != r1) return {-1, -1};
			continue;
		}
		int64_t g, im;
		tie(g, im) = inv_gcd(m0, m1);
		int64_t u1 = (m1/g);
		if((r1-r0)%g) return {-1, -1};
		int64_t x = (r1-r0)/g%u1 * im % u1;
		r0 += x*m0;
		m0 *= u1;
		if(r0 < 0) r0 += m0;
	}
	return {r0, m0};
}

//[0, n) (a*i+b)/m
int64_t floor_sum(int64_t n, int64_t m, int64_t a, int64_t b) {
	int64_t ans = 0;
	if(a >= m) {
		ans += ((n-1)*n>>1)*(a/m);
		a %= m;
	}
	if(b >= m) {
		ans += n*(b/m);
		b %= m;
	}
	int64_t y_max = (a*n+b)/m, x_max = y_max*m-b;
	if(y_max == 0) return ans;
	ans += (n - (x_max+a-1)/a) * y_max;
	ans += floor_sum(y_max, a, m, (a - x_max%a)%a);
	return ans;
}

//head

ll ans(pair<ll, ll> x) {
	return x.first?x.first:x.second;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	vector<ll> r(3), m(3);
	rep(i, 3) cin >> r[i] >> m[i];
	cout << ans(crt(r, m)) << endl;
}
0