結果
| 問題 |
No.2941 Sigma Music Game Score Problem
|
| ユーザー |
 asmin
|
| 提出日時 | 2025-03-18 20:15:17 |
| 言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 289 ms / 2,500 ms |
| コード長 | 6,431 bytes |
| コンパイル時間 | 2,397 ms |
| コンパイル使用メモリ | 210,608 KB |
| 実行使用メモリ | 12,928 KB |
| 最終ジャッジ日時 | 2025-03-18 20:15:27 |
| 合計ジャッジ時間 | 9,812 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 29 |
ソースコード
#pragma region header
#include<algorithm> // ソート, 二分探索, 最大・最小
#include<array> // 固定長配列（C++11~）
#include<bitset> // ビット演算・フラグ管理
#include<cassert> // デバッグ用の assert
#include<chrono> // 時間計測（C++11~）
#include<cinttypes> // int64_t, PRIu64 などのフォーマット指定
#include<climits> // INT_MAX, INT_MIN など
#include<cmath> // sqrt, pow, sin, cos など
#include<complex> // FFT（高速フーリエ変換）など
#include<cstdio> // printf, scanf（Cスタイル）
#include<cstring> // memset, memcpy
#include<deque> // 両端キュー（deque）
#include<functional> // 比較関数 (greater, less)
#include<iomanip> // 小数点表示の調整（setprecision など）
#include<iostream> // 標準入出力
#include<iterator> // イテレータ操作
#include<limits> // numeric_limits
#include<map> // std::map（連想配列）
#include<numeric> // gcd, lcm, accumulate など
#include<queue> // 優先度付きキュー (priority_queue)
#include<random> // 乱数生成（乱択アルゴリズムなど）
#include<set> // std::set（集合）
#include<sstream> // 文字列ストリーム（stringstream）
#include<stack> // スタック（LIFO）
#include<string> // 文字列操作
#include<tuple> // tuple, tie
#include<type_traits> // 型特性（C++11~）
#include<unordered_map> // ハッシュマップ（unordered_map）
#include<unordered_set> // ハッシュセット（unordered_set）
#include<utility> // pair, swap, move など
#include<vector> // 動的配列（最重要）
using namespace std;
struct Init{Init(){std::cin.tie(0); ios::sync_with_stdio(false); cout << setprecision(20) << fixed;}}init;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
#define all(x) begin((x)), end((x))
#define pb push_back
#define mp make_pair
#define mt make_tuple
#define uq(v) v.erase(unique(begin(v), end(v)), end(v))
#define _overload4(_1,_2,_3,_4,name,...) name
#define _overload3(_1,_2,_3,name,...) name
#define _rep1(n) for(int i=0;i<n;++i)
#define _rep2(i,n) for(int i=0;i<n;++i)
#define _rep3(i,a,b) for(int i=a;i<b;++i)
#define _rep4(i,a,b,c) for(int i=a;i<b;i+=c)
#define rep(...) _overload4(__VA_ARGS__,_rep4,_rep3,_rep2,_rep1)(__VA_ARGS__)
#define _rrep1(n) for(int i=(n)-1;i>=0;i--)
#define _rrep2(i,n) for(int i=(n)-1;i>=0;i--)
#define _rrep3(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define _rrep4(i,a,b,c) for(int i=a+(b-a-1)/c*c;i>=a;i-=c)
#define rrep(...) _overload4(__VA_ARGS__,_rrep4,_rrep3,_rrep2,_rrep1)(__VA_ARGS__)
template<class T> using pq = priority_queue<T>;
template<class T> using pq_g = priority_queue<T, vector<T>, greater<T>>;
template<class T> bool chmax(T &a, const T &b){if(a < b){a = b; return 1; } return 0;}
template<class T> bool chmin(T &a, const T &b){if(a > b){a = b; return 1; } return 0;}
template<class T> auto min(const T& a){ return *min_element(all(a)); }
template<class T> auto max(const T& a){ return *max_element(all(a)); }
constexpr ull INF = (1ULL << 61) + (1ULL << 30);
constexpr int inf = (1 << 30);
constexpr ld EPS = 1e-9;
constexpr ld PI = std::acos(ld(-1));
constexpr int dx[] = {1, 0, -1, 0, 1, 1, -1, -1};
constexpr int dy[] = {0, 1 ,0, -1, 1, -1, 1, -1};
#pragma endregion header
template <uint32_t mod_, bool fast = false>
struct MontgomeryModInt {
private:
using mint = MontgomeryModInt;
using i32 = int32_t;
using i64 = int64_t;
using u32 = uint32_t;
using u64 = uint64_t;
static constexpr u32 get_r() {
u32 ret = mod_;
for (i32 i = 0; i < 4; i++) ret *= 2 - mod_ * ret;
return ret;
}
static constexpr u32 r = get_r();
static constexpr u32 n2 = -u64(mod_) % mod_;
static_assert(r * mod_ == 1, "invalid, r * mod != 1");
static_assert(mod_ < (1 << 30), "invalid, mod >= 2 ^ 30");
static_assert((mod_ & 1) == 1, "invalid, mod % 2 == 0");
u32 x;
public:
MontgomeryModInt() : x{} {}
MontgomeryModInt(const i64 &a): x(reduce(u64(fast ? a : (a % mod() + mod())) * n2)) {}
static constexpr u32 reduce(const u64 &b) {
return u32(b >> 32) + mod() - u32((u64(u32(b) * r) * mod()) >> 32);
}
mint &operator+=(const mint &p) {
if (i32(x += p.x - 2 * mod()) < 0) x += 2 * mod();
return *this;
}
mint &operator-=(const mint &p) {
if (i32(x -= p.x) < 0) x += 2 * mod();
return *this;
}
mint &operator*=(const mint &p) {
x = reduce(u64(x) * p.x);
return *this;
}
mint &operator/=(const mint &p) {
*this *= p.inv();
return *this;
}
mint operator-() const { return mint() - *this; }
mint operator+(const mint &p) const { return mint(*this) += p; }
mint operator-(const mint &p) const { return mint(*this) -= p; }
mint operator*(const mint &p) const { return mint(*this) *= p; }
mint operator/(const mint &p) const { return mint(*this) /= p; }
bool operator==(const mint &p) const {
return (x >= mod() ? x - mod() : x) == (p.x >= mod() ? p.x - mod() : p.x);
}
bool operator!=(const mint &p) const {
return (x >= mod() ? x - mod() : x) != (p.x >= mod() ? p.x - mod() : p.x);
}
u32 val() const {
u32 ret = reduce(x);
return ret >= mod() ? ret - mod() : ret;
}
mint pow(u64 n) const {
mint ret(1), mul(*this);
while (n > 0) {
if (n & 1) ret *= mul;
mul *= mul;
n >>= 1;
}
return ret;
}
mint inv() const { return pow(mod() - 2); }
friend ostream &operator<<(ostream &os, const mint &p) {
return os << p.val();
}
friend istream &operator>>(istream &is, mint &a) {
i64 t;
is >> t;
a = mint(t);
return is;
}
static constexpr u32 mod() { return mod_; }
};
template <uint32_t mod>
using modint = MontgomeryModInt<mod>;
using mint = modint<998244353>;
int main(){
ll M; cin >> M;
int N; cin >> N;
vector<ll> X;
X.pb(0);
rep(i, N){
ll x; cin >> x;
X.pb(x);
}
X.pb(M + 1);
mint ans = 0;
rep(i, N + 1){
mint d = X[i + 1] - X[i] - 1;
ans += d * (d + 1) * (d * 2 + 1) / 6;
}
cout << ans << "\n";
}