#717668 (C++17) No.1750 ラムドスウイルスの感染拡大-hard

提出ソース

結果

問題	No.1750 ラムドスウイルスの感染拡大-hard
ユーザー	ytqm3
提出日時	2021-11-19 21:34:12
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	201 ms / 2,000 ms
コード長	4,846 bytes
コンパイル時間	2,072 ms
コンパイル使用メモリ	213,976 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-06-10 08:04:12
合計ジャッジ時間	4,987 ms
ジャッジサーバーID （参考情報）	judge5 / judge1

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	1 ms 6,944 KB
testcase_04	AC	6 ms 6,944 KB
testcase_05	AC	1 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	1 ms 6,940 KB
testcase_08	AC	36 ms 6,944 KB
testcase_09	AC	36 ms 6,940 KB
testcase_10	AC	38 ms 6,944 KB
testcase_11	AC	28 ms 6,944 KB
testcase_12	AC	39 ms 6,944 KB
testcase_13	AC	37 ms 6,944 KB
testcase_14	AC	189 ms 6,940 KB
testcase_15	AC	194 ms 6,940 KB
testcase_16	AC	194 ms 6,944 KB
testcase_17	AC	197 ms 6,944 KB
testcase_18	AC	201 ms 6,940 KB
testcase_19	AC	199 ms 6,944 KB
testcase_20	AC	137 ms 6,940 KB
testcase_21	AC	153 ms 6,944 KB
testcase_22	AC	27 ms 6,940 KB
testcase_23	AC	184 ms 6,940 KB
testcase_24	AC	22 ms 6,944 KB
testcase_25	AC	51 ms 6,940 KB
testcase_26	AC	17 ms 6,940 KB
testcase_27	AC	2 ms 6,940 KB
testcase_28	AC	2 ms 6,940 KB
testcase_29	AC	2 ms 6,940 KB
testcase_30	AC	27 ms 6,944 KB
testcase_31	AC	26 ms 6,944 KB
testcase_32	AC	25 ms 6,944 KB
testcase_33	AC	24 ms 6,940 KB

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

ソースコード

raw source code

#include<bits/stdc++.h>
typedef uint64_t u64;
typedef int64_t i64;
using namespace std;

void solve();

int main(){
  cout<<fixed<<setprecision(15);
  bool is_multicase=false;
  int T=1;
  if(is_multicase){
    cin>>T;
  }
  for(int i=0;i<T;++i){
    solve();
  }
  return 0;
}

template<typename T> void turtle_in(T& n){
  cin>>n;
}

template<typename T,typename U> void turtle_in(pair<T,U>& p){
  turtle_in(p.first);
  turtle_in(p.second);
}

template<typename T> void turtle_in(vector<T>& v){
  for(auto& u:v){
    turtle_in(u);
  }
}

template<typename T> void turtle_out(T& n){
  cout<<n;
}

template<typename T,typename U> void turtle_out(pair<T,U>& p){
  cout<<"{";
  turtle_out(p.first);
  cout<<", ";
  turtle_out(p.second);
  cout<<"}";
}

template<typename T> void turtle_out(vector<T>& v){
  for(int i=0;i<int(v.size());++i){
    turtle_out(v[i]);
    if(i!=v.size()-1){
      cout<<" ";
    }
  }
  cout<<endl;
}

template<typename T> void turtle_out(vector<vector<T>>& v){
  for(int i=0;i<int(v.size());++i){
    turtle_out(v[i]);
  }
}

void scan(){}

template<typename T,class... Args> void scan(T& n,Args&... args){
  turtle_in(n);
  scan(args...);
}

void print(){}

template<typename T,class... Args> void print(T n,Args... args){
  turtle_out(n);
  print(args...);
}

template<typename T> void println(T n){
  turtle_out(n);
  cout<<endl;
}

template<typename T,class... Args> void println(T n,Args... args){
  turtle_out(n);
  cout<<" ";
  println(args...);
}

template<u64 mod> struct modint{
  u64 val;
  modint(i64 val_=0):val((val_%i64(mod)+i64(mod))%i64(mod)){}
  modint operator-(){
    return modint(mod-val);
  }
  modint operator+(modint rhs){
    return modint(*this)+=rhs;
  }
  modint operator-(modint rhs){
    return modint(*this)-=rhs;
  }
  modint operator*(modint rhs){
    return modint(*this)*=rhs;
  }
  modint operator/(modint rhs){
    return modint(*this)/=rhs;
  }
  modint &operator+=(modint rhs){
    val+=rhs.val;
    val-=((val>=mod)?mod:0);
    return (*this);
  }
  modint &operator-=(modint rhs){
    val+=((val<rhs.val)?mod:0);
    val-=rhs.val;
    return (*this);
  }
  modint &operator*=(modint rhs){
    val=val*rhs.val%mod;
    return (*this);
  }
  modint &operator/=(modint rhs){
    u64 ex=mod-2;
    modint now=1;
    while(ex){
      now*=((ex&1)?rhs:1);
      rhs*=rhs,ex>>=1;
    }
    return (*this)*=now;
  }
  bool operator==(modint rhs){
    return val==rhs.val;
  }
  bool operator!=(modint rhs){
    return val!=rhs.val;
  }
  friend std::ostream &operator<<(std::ostream& os,modint x){
    return os<<(x.val);
  }
  friend std::istream &operator>>(std::istream& is,modint& x){
    u64 t;
    is>>t,x=t;
    return is;
  }
};

template<typename T,typename U> T power(T a,U n){
  T res=1;
  while(n>0){
    if(n%2==1){
      res*=a;
    }
    a*=a;
    n/=2;
  }
  return res;
}

template<typename T> struct genrand{
  uniform_int_distribution<T> gen;
  mt19937 engine;
  genrand(T a,T b):gen(a,b){
    random_device seed_gen;
    engine.seed(seed_gen());
  }
  T operator()(){
    return gen(engine);
  }
};

template<typename T> struct comb{
  vector<T> dat,idat;
  comb(int mx=3000000):dat(mx+1,1),idat(mx+1,1){
    for(int i=1;i<=mx;++i){
      dat[i]=dat[i-1]*i;
    }
    idat[mx]/=dat[mx];
    for(int i=mx;i>0;--i){
      idat[i-1]=idat[i]*i;
    }
  }
  T operator()(int n,int k){
    if(n<0||k<0||n<k){
      return 0;
    }
    return dat[n]*idat[k]*idat[n-k];
  }
};

template<typename T> struct matrix{
  vector<vector<T>> val;
  int height,width;
  matrix(int H,int W,T init=0):val(H,vector<T> (W,init)),height(H),width(W){}
  vector<T> &operator[](int i){
    return val[i];
  }
  matrix operator+(matrix rhs){
    return matrix(*this)+=rhs;
  }
  matrix operator*(matrix rhs){
    return matrix(*this)*=rhs;
  }
  matrix operator+=(matrix rhs){
    for(int i=0;i<rhs.height;++i){
      for(int j=0;j<rhs.width;++j){
        (*this)[i][j]+=rhs[i][j];
      }
    }
    return (*this);
  }
  matrix operator*=(matrix rhs){
    matrix<T> res(height,rhs.width,0);
    for(int i=0;i<this->height;++i){
      for(int j=0;j<rhs.width;++j){
        for(int k=0;k<this->width;++k){
          res[i][j]+=(*this)[i][k]*rhs[k][j];
        }
      }
    }
    return (*this)=res;
  }
  matrix identity(int n){
    matrix res(n,n);
    for(int i=0;i<n;++i){
      res[i][i]=1;
    }
    return res;
  }
  matrix mat_pow(i64 k){
    int n=(*this).height;
    matrix res=identity(n),now=(*this);
    while(k>0){
      if(k%2==1){
        res*=now;
      }
      now*=now;
      k/=2;
    }
    return (*this)=res;
  }
};

void solve(){
  typedef modint<998244353> mint;
  i64 N,M,T;
  scan(N,M,T);
  matrix<modint<998244353>> F(N,N),G(N,1);
  for(int i=0;i<M;++i){
    i64 s,t;
    scan(s,t);
    F[s][t]=1;
    F[t][s]=1;
  }
  G[0][0]=1;
  println((F.mat_pow(T)*G)[0][0]);
}

yukicoder

結果

テストケース

ソースコード