結果

問題 No.2017 Mod7 Parade
ユーザー siman
提出日時 2023-09-15 23:35:38
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
AC  
実行時間 74 ms / 2,000 ms
コード長 1,142 bytes
コンパイル時間 1,009 ms
コンパイル使用メモリ 140,392 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-03 00:50:23
合計ジャッジ時間 3,145 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

const ll MOD = 1000000007;

int main() {
  int K;
  cin >> K;
  vector<int> D(K);
  vector<int> L(K);

  for (int i = 0; i < K; i++) {
    cin >> D[i] >> L[i];
  }

  vector<int> mod7 = {1, 3, 2, 6, 4, 5};
  ll dp[6][7];
  memset(dp, 0, sizeof(dp));
  dp[0][0] = 1;

  for (int i = K - 1; i >= 0; --i) {
    ll temp[6][7];
    memcpy(temp, dp, sizeof(dp));

    int d = D[i];
    int l = L[i];
    for (int j = 0; j < 6; ++j) {
      ll sum = 0;
      for (int k = 0; k < l % 6; ++k) {
        sum += mod7[(j + k) % 6] * d;
      }
      for (int k = 0; k < 7; ++k) {
        temp[(j + l) % 6][(k + sum) % 7] += dp[j][k];
        temp[(j + l) % 6][(k + sum) % 7] %= MOD;
      }
    }

    memcpy(dp, temp, sizeof(dp));
  }

  ll ans = 0;
  for (int i = 0; i < 6; ++i) {
    for (int j = 0; j < 7; ++j) {
      ans += dp[i][j] * j;
      ans %= MOD;
    }
  }

  cout << ans << endl;

  return 0;
}
0