結果

問題 No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ
ユーザー tnakao0123
提出日時 2023-12-11 19:01:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 191 ms / 3,000 ms
コード長 1,591 bytes
コンパイル時間 759 ms
コンパイル使用メモリ 57,044 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-27 04:36:45
合計ジャッジ時間 4,170 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2581.cc:  No.2581 [Cherry Anniversary 3] 28輪の桜のブーケ - yukicoder
 */

#include<cstdio>
#include<vector>
#include<algorithm>
 
using namespace std;

/* constant */

const int N = 28;
const int K = N / 2;
const int KBITS = 1 << K;

/* typedef */

typedef vector<int> vi;

/* global variables */

int gs[N], hs[N];
vi bv0[K + 1], bv1[K + 1];

/* subroutines */

void addmod(int &a, int b, int m) { a = (a + b) % m; }

/* main */

int main() {
  int m;
  scanf("%d", &m);
  for (int i = 0; i < N; i++) scanf("%d", gs + i);
  for (int i = 0; i < N; i++) scanf("%d", hs + i);

  for (int bits = 0; bits < KBITS; bits++) {
    int bn = 0, s0 = 0, s1 = 0;
    for (int i = 0, bi = 1; i < K; i++, bi <<= 1) {
      bn += ((bits >> i) & 1);
      addmod(s0, (bits & bi) ? gs[i] : hs[i], m);
      addmod(s1, (bits & bi) ? gs[K + i] : hs[K + i], m);
    }

    bv0[bn].push_back(s0);
    bv1[bn].push_back(s1);
  }

  for (int i = 0; i <= K; i++) {
    sort(bv0[i].begin(), bv0[i].end());
    sort(bv1[i].begin(), bv1[i].end());
  }

  int qn;
  scanf("%d", &qn);

  while (qn--) {
    int ki, xi;
    scanf("%d%d", &ki, &xi);

    int cnt = 0;
    for (int i = 0, j = ki; j >= 0; i++, j--)
      if (i <= K && j <= K && ! bv0[i].empty() && ! bv1[j].empty())
	for (auto u: bv0[i]) {
	  auto vit0 = bv1[j].begin(), vit1 = bv1[j].end();
	  auto l0 = lower_bound(vit0, vit1, xi - u);
	  auto l1 = lower_bound(vit0, vit1, m - u);
	  auto l2 = lower_bound(vit0, vit1, m + xi - u);
	  cnt += (l1 - l0) + (vit1 - l2);
	}

    printf("%d\n", cnt);
  }

  return 0;
}
0