結果

問題 No.3182 recurrence relation’s intersection sum
ユーザー Cafe1942
提出日時 2025-05-30 17:55:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 597 ms / 2,000 ms
コード長 2,434 bytes
コンパイル時間 1,324 ms
コンパイル使用メモリ 122,940 KB
実行使用メモリ 9,472 KB
最終ジャッジ日時 2025-06-13 20:50:25
合計ジャッジ時間 11,674 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>//小数点出力用
//cout << fixed << setprecision(10) << ans;
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <unordered_set>
#include <set>
#include <map>
using ll = long long;
using namespace std;
#define modPHash (ll)((1LL<<61)-1)
#define modP (ll)998244353
bool chkrng0idx(int pos, int sup) { return (0 <= pos && pos < sup); }
int clk4(int num) { return (num - 2) * (num % 2); }
void yn(bool tf) { cout << (tf ? "Yes\n" : "No\n"); }
using ll = long long;

vector<vector<ll>> mul(vector<vector<ll>>& A, vector<vector<ll>>& B, int size) {
	vector<vector<ll>>res;
	for (int i = 0;i < size;i++) {
		vector<ll>row;
		for (int j = 0;j < size;j++) {
			ll tmp = 0;
			for (int k = 0;k < size;k++) {
				tmp += A[i][k] * B[k][j];
				tmp %= 998244353;
			}
			row.push_back(tmp);
		}
		res.push_back(row);
	}
	return res;
}


int main() {
	int K;
	cin >> K;
	ll L, R;
	cin >> L >> R;
	if (R == 0) {
		cout << 1;
		return 0;
	}
	vector<vector<ll>>M[64];
	vector<ll>U;
	for (int i = 0;i <= K;i++) {
		M[0].push_back(U);
		M[0][i].push_back(1);
		for (int j = 1;j <= i;j++) {
			M[0][i].push_back((M[0][i - 1][j - 1] + M[0][i - 1][j]) % 998244353);
		}
		for (int j = i + 1;j < K + 4;j++) {
			M[0][i].push_back(0);
		}
	}
	for (int i = K + 1;i < K + 4;i++) {
		M[0].push_back(U);
		for (int j = 0;j < K + 4;j++) {
			M[0][i].push_back(0);
		}
	}
	M[0][K + 1][K + 1] = K;
	M[0][K + 2][K] = 1;
	M[0][K + 2][K + 1] = 1;
	M[0][K + 2][K + 2] = K;
	M[0][K + 3][K] = 1;
	M[0][K + 3][K + 1] = 1;
	M[0][K + 3][K + 2] = K;
	M[0][K + 3][K + 3] = 1;
	for (int i = 1;i < 64;i++) {
		M[i] = mul(M[i - 1], M[i - 1], K + 4);
	}
	vector<vector<ll>>E, tmp;
	for (int i = 0;i < K + 4;i++) {
		E.push_back(U);
		for (int j = 0;j < K + 4;j++) {
			E[i].push_back(0);
		}
		E[i][i] = 1;
	}
	tmp = E;
	for (int i = 0;i < 64;i++) {
		if ((R >> i) & 1) {
			tmp = mul(tmp, M[i], K + 4);
		}
	}
	if (L == 0) {
		cout << (tmp[K + 3][0] + tmp[K + 3][K + 1] + tmp[K + 3][K + 2] + tmp[K + 3][K + 3]) % 998244353;
		return 0;
	}
	ll ans = (tmp[K + 3][0] + tmp[K + 3][K + 1] + tmp[K + 3][K + 2] + tmp[K + 3][K + 3]) % 998244353 + 998244353;
	tmp = E;
	for (int i = 0;i < 64;i++) {
		if (((L - 1LL) >> i) & 1) {
			tmp = mul(tmp, M[i], K + 4);
		}
	}
	cout << (ans - (tmp[K + 3][0] + tmp[K + 3][K + 1] + tmp[K + 3][K + 2] + tmp[K + 3][K + 3]) % 998244353) % 998244353;
	return 0;
}
0