#824644 (C++17) No.2159 Filling 4x4 array

提出ソース

結果

問題	No.2159 Filling 4x4 array
ユーザー	snuke
提出日時	2022-12-11 03:39:17
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	3,814 ms / 5,000 ms
コード長	6,399 bytes
コンパイル時間	2,811 ms
コンパイル使用メモリ	231,548 KB
実行使用メモリ	17,792 KB
最終ジャッジ日時	2024-10-15 03:09:18
合計ジャッジ時間	76,959 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	36 ms 5,632 KB
testcase_01	AC	36 ms 5,376 KB
testcase_02	AC	36 ms 5,504 KB
testcase_03	AC	37 ms 5,504 KB
testcase_04	AC	2,541 ms 14,336 KB
testcase_05	AC	38 ms 5,504 KB
testcase_06	AC	43 ms 5,504 KB
testcase_07	AC	36 ms 5,504 KB
testcase_08	AC	44 ms 5,632 KB
testcase_09	AC	37 ms 5,504 KB
testcase_10	AC	38 ms 5,376 KB
testcase_11	AC	40 ms 5,504 KB
testcase_12	AC	38 ms 5,504 KB
testcase_13	AC	43 ms 5,504 KB
testcase_14	AC	47 ms 5,632 KB
testcase_15	AC	41 ms 5,504 KB
testcase_16	AC	38 ms 5,632 KB
testcase_17	AC	39 ms 5,376 KB
testcase_18	AC	39 ms 5,504 KB
testcase_19	AC	41 ms 5,376 KB
testcase_20	AC	36 ms 5,632 KB
testcase_21	AC	51 ms 5,632 KB
testcase_22	AC	37 ms 5,504 KB
testcase_23	AC	42 ms 5,632 KB
testcase_24	AC	40 ms 5,632 KB
testcase_25	AC	3,358 ms 16,384 KB
testcase_26	AC	3,572 ms 17,152 KB
testcase_27	AC	3,780 ms 17,792 KB
testcase_28	AC	3,545 ms 17,024 KB
testcase_29	AC	3,611 ms 16,896 KB
testcase_30	AC	3,814 ms 17,792 KB
testcase_31	AC	3,567 ms 16,896 KB
testcase_32	AC	2,951 ms 15,488 KB
testcase_33	AC	3,615 ms 17,408 KB
testcase_34	AC	3,483 ms 16,640 KB
testcase_35	AC	3,179 ms 16,000 KB
testcase_36	AC	2,909 ms 15,360 KB
testcase_37	AC	3,167 ms 15,872 KB
testcase_38	AC	3,571 ms 17,024 KB
testcase_39	AC	2,926 ms 15,232 KB
testcase_40	AC	3,553 ms 16,768 KB
testcase_41	AC	3,439 ms 16,768 KB
testcase_42	AC	3,462 ms 16,768 KB
testcase_43	AC	3,424 ms 16,640 KB
testcase_44	AC	3,627 ms 17,152 KB
testcase_45	AC	68 ms 5,888 KB
testcase_46	AC	36 ms 5,504 KB
testcase_47	AC	127 ms 6,272 KB
testcase_48	AC	36 ms 5,376 KB
testcase_49	AC	139 ms 6,272 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define rep1(i,n) for(int i = 1; i <= (n); ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
#define srep(i,s,t) for (int i = s; i < (t); ++i)
#define rng(a) a.begin(),a.end()
#define rrng(a) a.rbegin(),a.rend()
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define pob pop_back
#define sz(x) (int)(x).size()
#define pcnt __builtin_popcountll
#define snuke srand((unsigned)clock()+(unsigned)time(NULL));
#define newline puts("")
using namespace std;
template<typename T> using vc = vector<T>;
template<typename T> using vv = vc<vc<T>>;
template<typename T> using PQ = priority_queue<T,vc<T>,greater<T>>;
using uint = unsigned; using ull = unsigned long long;
using vi = vc<int>; using vvi = vv<int>; using vvvi = vv<vi>;
using ll = long long; using vl = vc<ll>; using vvl = vv<ll>; using vvvl = vv<vl>;
using P = pair<int,int>; using vp = vc<P>; using vvp = vv<P>;
int geti(){int x;scanf("%d",&x);return x;}
vi pm(int n, int s=0) { vi a(n); iota(rng(a),s); return a;}
template<typename T>istream& operator>>(istream&i,vc<T>&v){rep(j,sz(v))i>>v[j];return i;}
template<typename T>string join(const T&v,const string& d=""){stringstream s;rep(i,sz(v))(i?s<<d:s)<<v[i];return s.str();}
template<typename T>ostream& operator<<(ostream&o,const vc<T>&v){if(sz(v))o<<join(v," ");return o;}
template<typename T1,typename T2>istream& operator>>(istream&i,pair<T1,T2>&v){return i>>v.fi>>v.se;}
template<typename T1,typename T2>ostream& operator<<(ostream&o,const pair<T1,T2>&v){return o<<v.fi<<","<<v.se;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int){a.fi--;a.se--;}
template<typename T>void operator--(vc<T>&a,int){for(T&x:a)x--;}
template<typename T>void operator++(vc<T>&a,int){for(T&x:a)x++;}
template<typename T>void operator+=(vc<T>&a,T b){for(T&x:a)x+=b;}
template<typename T>void operator+=(vc<T>&a,const vc<T>&b){a.insert(a.end(),rng(b));}
template<typename T1,typename T2>bool mins(T1& x,const T2&y){if(x>y){x=y;return true;}else return false;}
template<typename T1,typename T2>bool maxs(T1& x,const T2&y){if(x<y){x=y;return true;}else return false;}
template<typename Tx, typename Ty>Tx dup(Tx x, Ty y){return (x+y-1)/y;}
template<typename T>ll suma(const vc<T>&a){ll res(0);for(auto&&x:a)res+=x;return res;}
template<typename T>ll suma(const vv<T>&a){ll res(0);for(auto&&x:a)res+=suma(x);return res;}
template<typename T>void uni(T& a){sort(rng(a));a.erase(unique(rng(a)),a.end());}
template<typename T>void prepend(vc<T>&a,const T&x){a.insert(a.begin(),x);}
const double eps = 1e-10;
const ll LINF = 1001002003004005006ll;
const int INF = 1001001001;
#define dame { puts("-1"); return;}
#define yes { puts("Yes"); return;}
#define no { puts("No"); return;}
#define ret(x) { cout<<(x)<<endl; return;}
#define yn {puts("Yes");}else{puts("No");}
const int MX = 200005;

// Mod int
const uint mod = 998244353;
struct mint {
  uint x;
  mint(): x(0) {}
  mint(ll x):x((x%mod+mod)%mod) {}
  mint operator-() const { return mint(0) - *this;}
  mint operator~() const { return mint(1) / *this;}
  mint& operator+=(const mint& a) { if((x+=a.x)>=mod) x-=mod; return *this;}
  mint& operator-=(const mint& a) { if((x+=mod-a.x)>=mod) x-=mod; return *this;}
  mint& operator*=(const mint& a) { x=(ull)x*a.x%mod; return *this;}
  mint& operator/=(const mint& a) { x=(ull)x*a.pow(mod-2).x%mod; return *this;}
  mint operator+(const mint& a) const { return mint(*this) += a;}
  mint operator-(const mint& a) const { return mint(*this) -= a;}
  mint operator*(const mint& a) const { return mint(*this) *= a;}
  mint operator/(const mint& a) const { return mint(*this) /= a;}
  mint pow(ll t) const {
    mint res = 1; for (mint p=x;t;p*=p,t>>=1) if (t&1) res *= p; return res;
  }
  mint ppow(ll t) const { int p=mod-1; return pow((t%p+p)%p);}
  bool operator<(const mint& a) const { return x < a.x;}
  bool operator==(const mint& a) const { return x == a.x;}
  bool operator!=(const mint& a) const { return x != a.x;}
};
mint ex(mint x, ll t) { return x.pow(t);}
istream& operator>>(istream&i,mint&a) {i>>a.x;return i;}
//*
ostream& operator<<(ostream&o,const mint&a) {o<<a.x;return o;}
/*/
ostream& operator<<(ostream&o, const mint&x) {
  int a = x.x, b = 1;
  rep(s,2)rep1(i,1000) {
    int y = ((s?-x:x)*i).x; if (abs(a)+b > y+i) a = s?-y:y, b = i;
  }
  o<<a; if (b != 1) o<<'/'<<b; return o;
}//*/
using vm = vector<mint>;
using vvm = vector<vm>;
struct modinv {
  int n; vm d;
  modinv(): n(2), d({0,1}) {}
  mint operator()(int i) { while (n <= i) d.pb(-d[mod%n]*(mod/n)), ++n; return d[i];}
  mint operator[](int i) const { return d[i];}
} invs;
struct modfact {
  int n; vm d;
  modfact(): n(2), d({1,1}) {}
  mint operator()(int i) { while (n <= i) d.pb(d.back()*n), ++n; return d[i];}
  mint operator[](int i) const { return d[i];}
} facs;
struct modfactinv {
  int n; vm d;
  modfactinv(): n(2), d({1,1}) {}
  mint operator()(int i) { while (n <= i) d.pb(d.back()*invs(n)), ++n; return d[i];}
  mint operator[](int i) const { return d[i];}
} ifacs;
mint comb(int a, int b) {
  if (a < b || b < 0) return 0;
  return facs(a)*ifacs(b)*ifacs(a-b);
}
//

struct Solver {
  void solve() {
    int n = 4, n2 = n*2;
    vi d(n2); cin>>d; d += -n;
    int m = n*n, m2 = 1<<m;
    vc<map<vi,int>> ax(1<<n2);
    rep(s,m2) {
      vvi b(n, vi(n));
      rep(i,n)rep(j,n) b[i][j] = s>>(i*n+j)&1;
      vi na;
      rep(i,n) na.pb(suma(b[i]));
      rep(i,n) {
        na.pb(0);
        rep(j,n) na.back() += b[j][i];
      }
      int ns = 0;
      rep(i,n2) ns |= (na[i]&1)<<i;
      ax[ns][na]++;
    }

    map<vi,mint> dp;
    dp[vi(n2)] = 1;
    auto f = [&](auto f, const vi& d) -> mint {
      auto it = dp.find(d);
      if (it != dp.end()) return it->se;
      int ns = 0;
      rep(i,n2) ns |= (d[i]&1)<<i;
      mint res;
      for (auto [a,x] : ax[ns]) {
        // cerr<<a<<" * "<<x<<endl;
        vi nd = d;
        rep(i,n2) {
          if (nd[i] < a[i]) { nd = vi(); break;}
          nd[i] -= a[i]; nd[i] >>= 1;
        }
        if (sz(nd)) res += f(f,nd)*x;
      }
      // cerr<<d<<": "<<res<<endl;
      return dp[d] = res;
    };

    mint ans = f(f,d);
    cout<<ans<<endl;
  }
};

int main() {
  // cin.tie(nullptr); ios::sync_with_stdio(false);
  int ts = 1;
  // scanf("%d",&ts);
  rep1(ti,ts) {
    Solver solver;
    solver.solve();
  }
  return 0;
}

yukicoder

結果

テストケース

ソースコード