結果

問題 No.213 素数サイコロと合成数サイコロ (3-Easy)
ユーザー okadukiokaduki
提出日時 2016-03-15 01:38:57
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 751 ms / 3,000 ms
コード長 2,109 bytes
コンパイル時間 1,959 ms
コンパイル使用メモリ 177,304 KB
実行使用メモリ 6,548 KB
最終ジャッジ日時 2024-04-08 13:14:58
合計ジャッジ時間 3,806 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 284 ms
6,548 KB
testcase_01 AC 751 ms
6,548 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef long long LL;
typedef pair<LL, LL> PLL;

#define ALL(a)  (a).begin(),(a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define SZ(a) int((a).size())
#define EACH(i,c) for(typeof((c).begin()) i=(c).begin(); i!=(c).end(); ++i)
#define EXIST(s,e) ((s).find(e)!=(s).end())
#define SORT(c) sort((c).begin(),(c).end())

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n)  FOR(i,0,n)

#define FF first
#define SS second
template<class S, class T>
istream& operator>>(istream& is, pair<S,T>& p){
  return is >> p.FF >> p.SS;
}

const double EPS = 1e-10;
const double PI  = acos(-1.0);
const LL MOD = 1e9+7;

void dfs(VI& xs, map<int,int>& res, int n, int i=0, int acc = 0){
  if(n == 0){
	res[acc]++;
	return;
  }
  if(i >= 6) return;
  for(int m=0;m<=n;++m)
	dfs(xs, res, n-m, i+1, acc+xs[i]*m);
}

typedef vector<LL> Col;
typedef vector<Col> Matrix;

Matrix mul(const Matrix& A, const Matrix& B){
  const int R = A.size(), C = B[0].size(), sz = B.size();
  Matrix AB(R, Col(C));

  for(int i=0;i<R;++i)
    for(int j=0;j<C;++j)
      for(int k=0;k<sz;++k)
		(AB[i][j] += A[i][k] * B[k][j]) %= MOD;

  return AB;
}

Matrix powA(const Matrix& A, LL n){
  const int N = A.size();
  Matrix p(N, Col(N, 0)), w = A;
  for(int i=0;i<N;++i) p[i][i] = 1;

  while(n>0){
	if(n&1)
	  p = mul(p, w);
	w = mul(w, w);
	n >>= 1;
  }

  return p;
}


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

  VI ps{2,3,5,7,11,13};
  VI qs{4,6,8,9,10,12};
  LL N, P, C; cin >> N >> P >> C;
  map<int,int> memo1, memo2;
  dfs(ps, memo1, P);
  dfs(qs, memo2, C);

  map<int,int> pat;
  for(auto& p1: memo1)
	for(auto& p2: memo2)
	  pat[p1.FF+p2.FF] += p1.SS * p2.SS;

  int D = 13*P + 12*C;
  Matrix p(D, Col(D));
  REP(i,D-1) p[i][i+1] = 1;
  for(auto& x: pat)
	p[x.FF-1][0] = x.SS;
  p = powA(p, N);

  LL ans = 0;
  REP(i,D) (ans += p[i][0]) %= MOD;

  cout << ans << endl;

  return 0;
}
0