結果

問題 No.1348 Split Tile
ユーザー kk
提出日時 2021-04-18 17:53:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 357 ms / 2,000 ms
コード長 2,703 bytes
コンパイル時間 2,028 ms
コンパイル使用メモリ 196,432 KB
最終ジャッジ日時 2025-01-20 21:46:01
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;

template<typename T,T MOD>
struct mod_int{
  T v;
  mod_int(): v(0) {}
  mod_int(T t) { v=t%MOD; if(v<0) v+=MOD; }
  
  mod_int inv() const {
    T a=v, b=MOD, u=1, v=0;
		while (b>0) { T t=a/b; a-=t*b; swap(a,b); u-=t*v; swap(u,v); }
		return u;
  }
  
  mod_int pow(long long p) const {
    mod_int ret = 1, base(v);
    while (p > 0) {
      if (p & 1)
        ret *= base;
      base *= base;
      p >>= 1;
    }
    return ret;
  }
  
  mod_int& operator+=(const mod_int &a) { v+=a.v; if(v>=MOD)v-=MOD; return *this; }
  mod_int& operator-=(const mod_int &a) { v+=MOD-a.v; if(v>=MOD)v-=MOD; return *this; }
  mod_int& operator*=(const mod_int &a) { v=v*a.v%MOD; return *this; }
  mod_int& operator/=(const mod_int &a) { return (*this)*=a.inv(); }
  
  mod_int operator+(const mod_int &a) const { return mod_int(v)+=a; }
  mod_int operator-(const mod_int &a) const { return mod_int(v)-=a; }
  mod_int operator*(const mod_int &a) const { return mod_int(v)*=a; }
  mod_int operator/(const mod_int &a) const { return mod_int(v)/=a; }
  
  friend mod_int operator+(T x, const mod_int& m) { return mod_int(x)+m; }
  friend mod_int operator-(T x, const mod_int& m) { return mod_int(x)-m; }
  friend mod_int operator*(T x, const mod_int& m) { return mod_int(x)*m; }
  friend mod_int operator/(T x, const mod_int& m) { return mod_int(x)/m; }
  
  mod_int operator-() const{ return v?mod_int(MOD-v):mod_int(v); }
  
  bool operator==(const mod_int &a) const { return v==a.v; }
  bool operator!=(const mod_int &a) const { return v!=a.v; }
  bool operator<(const mod_int &a) const { return v <a.v; }
  
	friend istream& operator>>(istream& is, mod_int& m){ T t; is>>t; m=mod_int(t); return is; }
	friend ostream& operator<<(ostream& os,const mod_int& m){ return os<<m.v; }
};

template<typename T>
class chooser {
  vector<T> f, g;
public:
  chooser(int n) {
    f.resize(n + 1);
    g.resize(n + 1);
    
    f[0] = g[0] = 1;
    for (int i = 1; i <= n; i++) {
      f[i] = i * f[i-1];
      g[i] = 1 / f[i];
    }
  }

  T perm(int n, int k) {
    return k > n ? 0 : f[n] * g[n-k];
  }
  
  T comb(int n, int k) {
    return k > n ? 0 : f[n] * g[k] * g[n-k];
  }
  
  T homo(int n, int k) {
    return comb(n+k-1, k);
  }
};

using mint = mod_int<long long, 998244353>;

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  int n;
  cin >> n;

  mint ret;
  chooser<mint> choo(n);
  
  // 先頭
  for (int i = 1; i < n; i++)
    ret += choo.perm(n-1, i) * choo.perm(n-i, n-i);
  
  // 先頭以外
  for (int i = 1; i < n; i++)
    ret += choo.comb(i, 1) * choo.perm(n-2, i-1) * choo.perm(n-i, n-i) * (n-1);

  cout << ret << endl;
  
  return 0;
}
0