#1005325 (C++23) No.2857 Div Array

提出ソース

結果

問題	No.2857 Div Array
ユーザー	sgfc
提出日時	2024-08-27 01:38:05
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	33 ms / 2,000 ms
コード長	4,290 bytes
コンパイル時間	7,888 ms
コンパイル使用メモリ	458,796 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-08-27 01:38:15
合計ジャッジ時間	7,851 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	1 ms 6,816 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	1 ms 6,940 KB
testcase_05	AC	2 ms 6,944 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	2 ms 6,940 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	3 ms 6,944 KB
testcase_11	AC	25 ms 6,940 KB
testcase_12	AC	7 ms 6,944 KB
testcase_13	AC	4 ms 6,940 KB
testcase_14	AC	24 ms 6,940 KB
testcase_15	AC	4 ms 6,940 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	AC	23 ms 6,944 KB
testcase_18	AC	19 ms 6,944 KB
testcase_19	AC	11 ms 6,944 KB
testcase_20	AC	32 ms 6,940 KB
testcase_21	AC	33 ms 6,944 KB
testcase_22	AC	33 ms 6,940 KB
testcase_23	AC	32 ms 6,944 KB
testcase_24	AC	2 ms 6,940 KB
testcase_25	AC	3 ms 6,940 KB
testcase_26	AC	2 ms 6,940 KB
testcase_27	AC	3 ms 6,940 KB
testcase_28	AC	1 ms 6,940 KB
testcase_29	AC	1 ms 6,944 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include "bits/stdc++.h"

#include <stdio.h>
#include <math.h>

#include "atcoder/all"
#include <float.h>

#include <boost/multiprecision/cpp_int.hpp>
using namespace boost::multiprecision;

using namespace std;
using namespace atcoder;

using ll = long long;
using ull = unsigned long long;
using mint = modint998244353;
using mint2 = modint1000000007;
const double eps = 1e-9;

#define REP(i, n) for (ll i = 0; i < ll(n); ++i)
#define REPI(i, init, n) for (ll i = init; i < ll(n); ++i)
#define REPD(i, init) for (ll i = init; i >=0; --i)
#define REPDI(i, init, n) for (ll i = init; i >=n; --i)
#define REPE(i, c) for (const auto& i : c)
#define TCASE() ll _ttt; cin >> _ttt; while(_ttt--)

using vl = vector<ll>; using vvl = vector<vl>; using vvvl = vector<vvl>; using dl = deque<ll>;
#define VLI(a, n) vector<ll> a; a.assign(n, 0); for(auto& x : a) cin >> x;
#define DLI(a, n) deque<ll> a; a.assign(n, 0); for(auto& x : a) cin >> x;
#define VSI(a, n) vector<string> a; a.assign(n, ""); for(auto& x : a) cin >> x;
#define VVLI(a, n, m, init) vector<vector<ll>> a(n); for(auto& x : a) x.assign(m, init);
using vd = vector<double>; using pl = pair<ll, ll>; using vpl = vector<pl>;

struct uv { ll u; ll v; ll c; };
#define VUVI(a, n) vector<uv> a; a.assign(n, {0, 0, 1}); for(auto& x : a) {cin >> x.u >> x.v; x.u--; x.v--;}
#define VUVCI(a, n) vector<uv> a; a.assign(n, {0, 0, 0}); for(auto& x : a) {cin >> x.u >> x.v >> x.c; x.u--; x.v--;}
vvl to_edge(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) ret[x.u].push_back(x.v); return ret; }
vvl to_edge_d(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) { ret[x.u].push_back(x.v); ret[x.v].push_back(x.u); } return ret; }

template <class T = long long> using pql = priority_queue<T>;
template <class T = long long> using pqg = priority_queue<T, vector<T>, greater<T>>;
using vm = vector<mint>; using vvm = vector<vm>;

ostream& operator<<(ostream& os, const mint& v) { os << v.val(); return os; }
void yn(bool f) { std::cout << (f ? "Yes" : "No") << endl; };
template<class T> void ov(const T& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << *it; } };
template<> void ov(const vm& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << it->val(); } };

const ll llhuge = 1LL << 62; //十分でかいが多少足しても溢れない数値
auto gmax(const auto& a, const auto& b) { return a > b ? a : b; };
auto gmin(const auto& a, const auto& b) { return a < b ? a : b; };
bool chmin(auto& a, const auto& b) { const auto olda = a; a = gmin(a, b); return olda != a; }
bool chmax(auto& a, const auto& b) { const auto olda = a; a = gmax(a, b); return olda != a; }
void sort(auto& v) { std::ranges::sort(v); };
void rev(auto& v) { std::ranges::reverse(v); };
template<class T> auto p_sum(const T& v) { T ret(v.size()); std::partial_sum(begin(v), end(v), ret.begin()); return ret; };
template<class T, class K = T::key_type, class V = T::mapped_type> [[nodiscard]] V myat(const T& mp, const K& key) { return mp.find(key) == mp.end() ? V() : mp.at(key); }
long long ceil(const long long a, const long long b) { return (a + b - 1) / b; }


using mat_row = vector<mint>;
using mat = vector<mat_row>;
mat make_mat(ll size, mint init = 0) {
	return mat(size, mat_row(size, init));
}

mat operator*(const mat& a, const mat& b) {
	mat c(a.size(), mat_row(b[0].size()));
	for (size_t i = 0; i < c.size(); ++i)
		for (size_t j = 0; j < c[0].size(); ++j)
			for (size_t k = 0; k < a[0].size(); ++k)
				c[i][j] += a[i][k] * b[k][j];
	return c;
}
mat pow(mat a, ll n) {
	mat res(a.size(), mat_row(a[0].size()));
	for (size_t i = 0; i < res.size(); ++i) res[i][i] = 1;
	for (; n; n /= 2) {
		if (n & 1) res = res * a;
		a = a * a;
	}
	return res;
}


int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	ll n, m, k;
	cin >> n >> m >> k;

	map<ll, ll> mp;
	REP(i, m) mp[m / (i + 1)] = 0;
	ll cnt = 0;
	for(auto& x : mp) x.second = cnt++;
	auto mt = make_mat(cnt, mint(0));

	REPE(e, mp) {
		REP(j, m) {
			if (abs(e.first - m / (j + 1)) <= k) {
				mt[e.second][mp[m / (j + 1)]] += 1;
			}
		}
	}

	auto pmt = pow(mt, n - 1);
	mint ans = 0;
	REP(i, m) {
		REPE(e, pmt[mp[m / (i + 1)]]) ans += e;
	}

	cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード