結果

問題 No.1683 Robot Guidance
ユーザー rtyu
提出日時 2021-10-13 15:16:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 48 ms / 2,000 ms
コード長 1,471 bytes
コンパイル時間 775 ms
コンパイル使用メモリ 72,240 KB
実行使用メモリ 50,304 KB
最終ジャッジ日時 2024-09-17 16:17:47
合計ジャッジ時間 2,870 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;

const long long md = 1000000007;
long long ft[4000009], rt[4000009];

long long fp(long long n, long long k)
{
	long long s = 1;
	while (k) {
		if (k & 1) s = (s * n) % md;
		n = (n * n) % md; k /= 2;
	}
	return s;
}

long long cb(int n, int k)
{
	if (n < k) return 0;
	return (((ft[n] * rt[k]) % md) * rt[n - k]) % md;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int a, b, x, y; cin >> a >> b >> x >> y;
	if (((a + abs(x) + abs(y)) & 1) || abs(x) + abs(y) > a) {
		cout << 0 << '\n';
		return 0;
	}
	if (!a) {
		cout << 1 << '\n';
		return 0;
	}
	if (!b) {
		if (x == a && !y) cout << 1 << '\n';
		else cout << 0 << '\n';
		return 0;
	}
	int n = a + b + 1 + abs(x) + abs(y);
	ft[0] = rt[0] = 1;
	for (int i = 1; i <= n; i++)
		ft[i] = (ft[i - 1] * i) % md;
	rt[n] = fp(ft[n], md - 2);
	for (int i = n - 1; i >= 1; i--)
		rt[i] = (rt[i + 1] * (i + 1)) % md;
	int c[4] = { 0, 0, 0, 0 }, r[4] = { 0, 0, 0, 0 }, tp = 0;
	for (int i = 0; i <= b; i++) {
		c[tp]++;
		tp = (tp + 1) % 4;
	}
	if (x > 0) r[0] = x;
	else r[2] = -x;
	if (y > 0) r[1] = y;
	else r[3] = -y;
	a = (a - (abs(x) + abs(y))) / 2;
	long long ans = 0;
	r[1] += a; r[3] += a;
	for (int i = 0; i <= a; i++) {
		long long s = 1;
		for (int j = 0; j < 4; j++) 
			if (r[j])
				s = (s * cb(c[j] + r[j] - 1, r[j])) % md;
		ans = (ans + s) % md;
		r[0]++; r[1]--; r[2]++; r[3]--;
	}
	cout << ans << '\n';
	return 0;
}
0