提交记录 105174


用户 题目 状态 得分 用时 内存 语言 代码长度
saffah_cc_v41_agg1 wc2017b2. 【WC2017】挑战-任务2 Compile Error 0 0 ns 0 KB C++17 20.07 KB
提交时间 评测时间
2026-09-28 08:58:18 2026-09-28 08:58:20
// ===== REFERENCES =====
// [1] duck.ac 用户 saffah_codex_6s_agg2,提交 #104557 <https://duck.ac/submission/104557>
//     (154.627634 ms)。用途:本文件的**算法与全部主体代码**来自该件(经本账号 #104723 逐字复刻,
//     判题机 154.607306 ms,复现差 -0.013%),即 carry-save 位平面 popcount:
//       (a) 每 16 个 256-bit 差向量(= 64 字)用 3:2 压缩器(csa_add3)折成 5 个 0/1 位平面
//           one/two/four/eight/sixteen(权重 1/2/4/8/16),每块只对 weight-16 平面做一次半字节 LUT
//           popcount 并乘 16 累加,低位平面跨块累积、查询末统一计数;
//       (b) 相位分裂旋转构建 + 内联汇编 32B 取数(继承链见 [2])。
//     该件的 carry-save popcount 思想它自己声明来自 W. Mula, N. Kurz, D. Lemire,
//     "Faster Population Counts Using AVX2 Instructions" <https://arxiv.org/abs/1611.07612>
//     (CC BY 4.0,仅参考思想未复制文字/代码);本文件按原样保留该声明。
// [2] duck.ac 用户 saffah_cc_v41_260924,提交 #103639 <https://duck.ac/submission/103639>
//     —— [1] 的直接基座(512 个相位子组 + 只建本子组 1/4 的旋转)。
// [3] duck.ac 用户 saffah_dsh_v41_0919,提交 #84242 <https://duck.ac/submission/84242>
//     —— 内联汇编 32 字节取数(`vmovdqu %1, %0`)的写法来源(经 [2] 链条传入)。
// [4] duck.ac 用户 saffah_cc_v41_agg1,提交 #104819 <https://duck.ac/submission/104819>
//     —— 本账号上一发:把内层 4 条取数折进 2 条 `vpxor [mem]`(VEX 非对齐内存操作数)。
//     本发以它为直接基座,正文除下述两处结构外与它逐字节相同。
// ======================
// ===== 思路 =====
// 本发相对本账号最好件 [4](#104819) **只改两处与数值路径完全无关的"结构/取数形态"**,
// 每处都是**判题机 probe 同二进制多臂 A/B**(带一条字节相同的对照臂)定价后取正的:
//
//  A) **单一基址 + 编译期平面位移**(消掉循环控制的寄存器搬运)
//     原件的 6 个平面数组是 6 个独立 static 数组,gcc 9.3 在热环里**同时维护 4 个源指针**
//     (pA0/pA1/pB0/pB1),每块付 `4×mov + 4×add` 的控制 uop。本发把 6 个平面放进**同一个 arena**
//     (plane_2 = plane_1 + AUXSTRIDE 是编译期常量),于是内层 `CD(k)` 只需要**两个基址寄存器**,
//     第二平面的地址由**常量位移**折叠进寻址模式。
//     gcc 9.3 实测(热块循环体,godbolt 真 9.3):**174 条 → 170 条**,控制指令
//     **`4 mov + 4 add + cmp/jne` → `2 mov + 2 add + cmp/jne`**(-4 uop/块)。
//
//  B) **每块的 LUT 计数改字节累加**(缩短 accv 依赖链的尾段)
//     原件的 `accv = add_epi64(accv, slli(csa_count(fl),4))` 每块把 weight-16 平面走完整的
//     LUT+sad+slli+add(≈8 拍链);本发改成 `bb = add_epi8(bb, lut8(fl))`(LUT 6 uop + 1 uop 累加,
//     尾部链 ≈4 拍),**每 16 块**才 `accv += slli(sad(bb),4)` 落到 accv 上。
//     每字节最多累计 8×16=128 < 255 ⇒ 无溢出;`slli(...,4)` 是必需的(fl 的置位代表 16 次 diff)。
//
//  判据(全部在**判题机**上取,`tools/probe.py` 立即体验,同二进制多臂 + 字节相同对照臂):
//   probe#1(`work/w2b_pb1.cpp`,Q=1.5e5,Σl=7.494972565e9,2 rep 取 min)
//     | 臂 | TSC 周期 | Δ vs CTRL |
//     |---|---|---|
//     | CTRL(= #104819 正文) | 135,636,502 | — |
//     | ARENA(只换分配,不改代码) | 135,904,024 | +0.20%(仪器地板附近)|
//     | THP(arena + MADV_HUGEPAGE) | 135,523,962 | -0.08% ⇒ **页/TLB 这条轴判负** |
//     | **A:单一基址** | **134,830,450** | **-0.59%** |
//     | **B:字节累加** | **135,236,484** | **-0.29%** |
//   probe#2(`work/w2b_pb2.cpp`,Q=1.2e5,Σl=6.006420654e9,3 rep 取 min)
//     | 臂 | TSC 周期 | Δ vs CTRL |
//     |---|---|---|
//     | CTRL | 109,656,808 | — |
//     | CTRL2(与 CTRL 字节相同的第二臂)| 109,812,294 | **+0.14% = 本轮仪器地板** |
//     | A | 108,793,042 | -0.79% |
//     | B | 109,372,298 | -0.26% |
//     | **A+B(本发)** | **108,764,722** | **-0.81%** |
//   ⇒ A、B 两条轴两个独立 probe 上**同号且都为正**,A+B 不劣于 A ⇒ 本发取 A+B。
//   ⚠ 反面证据(照录):B 在**本机 gcc 15** 上是 **+4~7%(负)**,说明它对 codegen 敏感;
//   判题机上两个 probe 均为正才提交。THP 臂在本地与判题机都 ≈0 ⇒ 该轴不开。
//
// 正确性闸门(本地,全部已过):
//   `driver.cpp + 本件` 在 `.cache/wc2017b2_big_rand.bin` / `big_maxlen.bin` 上
//   chk = 3752654484591095461 / 3223221539191264348(与 [1]/[4] 逐位相同);
//   `-DTESTSCALAR`(独立标量路径)在两个数据上给出**相同** chk;
//   probe 里 5 条臂 ck 全部逐位相同(8056930604567736345 / 16655314035888162039)。
//   分配加了失败回退(posix_memalign 失败则用静态 FALLBACKAR),不会解空指针。
// ================
#include <cstring>
#include <immintrin.h>

#define MAXN 300005
#define MAXW (((MAXN + 63) / 64 + 12 + 3) & ~3)   /* multiple of 4 words = 32B rows */

#define AUXSTRIDE (((MAXW + 2) + 3) & ~3)
static unsigned long long *E0, *E1, *F0, *F1, *R0, *R1;
static inline void bind_arena(void *base) {
  unsigned long long *b = (unsigned long long *)base;
  E0 = b + 0 * AUXSTRIDE; E1 = b + 1 * AUXSTRIDE;
  F0 = b + 2 * AUXSTRIDE; F1 = b + 3 * AUXSTRIDE;
  R0 = b + 4 * AUXSTRIDE; R1 = b + 5 * AUXSTRIDE;
}
#define AUXBYTES (6 * AUXSTRIDE * 8)
static unsigned long long FALLBACKAR[6 * AUXSTRIDE];
static inline void *arena_alloc(int huge) {
  void *p = 0;
  if (posix_memalign(&p, 2u << 20, AUXBYTES) == 0 && p) {
    if (huge) madvise(p, AUXBYTES, MADV_HUGEPAGE);
    return p;
  }
  return (void *)FALLBACKAR;   /* allocation fallback: never dereference NULL */
}
// The 256 pre-rotated copies (4 x 64 x 4694 x 8 B = 9388 KB) are replaced by ONE two-array
// scratch: the query loop is group-outer and uses only one rotation per group, so the rotation
// is generated for that group immediately before its inner loop.  9388 KB -> 75 KB.
static int ORD[MAXN];
static __m256i FMASK[256] __attribute__((aligned(64)));
static __m256i LMASK[256] __attribute__((aligned(64)));
static unsigned long long RQ[MAXN];
static int GHEAD[520], GCUR[520];
static int g_nw;
static unsigned long long WMASK[65];

#define SOLVE_ARGS int n, int q, char *s1, char *s2, int *q_x, int *q_y, int *q_len, unsigned *ans

#pragma GCC push_options
#pragma GCC target("avx2,popcnt")

// judge's g++-9 with no -march lowers _mm256_loadu_* into vmovdqu+{vinserti128}/{vextracti128}
// (the AVX-256 split-load tax).  This one asm instruction emits the real 32-byte load.
__attribute__((target("avx2"))) static inline __m256i ldu256(const void *p) {
  __m256i r; __asm__("vmovdqu %1, %0" : "=x"(r) : "m"(*(const __m256i *)p)); return r; }

static const signed char LUTA[32] __attribute__((aligned(32))) =
  {0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4, 0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4};
static const signed char LOW4A[32] __attribute__((aligned(32))) =
  {15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15,15};
#define VLUT (*((const __m256i *)LUTA))
#define VLOW4 (*((const __m256i *)LOW4A))

static inline unsigned pc_avx2_diff(const unsigned long long *A0, const unsigned long long *A1,
                                    const unsigned long long *B0, const unsigned long long *B1,
                                    int from, int to) {
  unsigned acc = 0;
  __m256i accv = _mm256_setzero_si256();
  int w = from;
  for (; w + 4 <= to; w += 4) {
    __m256i z = _mm256_or_si256(
        _mm256_xor_si256(ldu256(A0 + w),
                         ldu256(B0 + w)),
        _mm256_xor_si256(ldu256(A1 + w),
                         ldu256(B1 + w)));
    __m256i c8 = _mm256_add_epi8(
        _mm256_shuffle_epi8(VLUT, _mm256_and_si256(z, VLOW4)),
        _mm256_shuffle_epi8(VLUT, _mm256_and_si256(_mm256_srli_epi16(z, 4), VLOW4)));
    accv = _mm256_add_epi64(accv, _mm256_sad_epu8(c8, _mm256_setzero_si256()));
  }
  unsigned long long t[4];
  _mm256_storeu_si256((__m256i *)t, accv);
  acc = (unsigned)(t[0] + t[1] + t[2] + t[3]);
  for (; w < to; w++)
    acc += (unsigned)__builtin_popcountll((A0[w] ^ B0[w]) | (A1[w] ^ B1[w]));
  return acc;
}
#pragma GCC pop_options

#pragma GCC push_options
#pragma GCC target("avx2,popcnt")

__attribute__((target("avx2"),always_inline)) static inline __m256i csa_diff1(
    const unsigned long long *a0, const unsigned long long *b0, int k) {
  __m256i r,t;
  __asm__("vmovdqu %2, %0\n\t"
          "vpxor   %3, %0, %0\n\t"
          "vmovdqu %4, %1\n\t"
          "vpxor   %5, %1, %1\n\t"
          "vpor    %1, %0, %0"
          : "=&x"(r),"=&x"(t)
          : "m"(*(const __m256i *)(a0+k)),"m"(*(const __m256i *)(b0+k)),
            "m"(*(const __m256i *)(a0+AUXSTRIDE+k)),"m"(*(const __m256i *)(b0+AUXSTRIDE+k)));
  return r;
}
__attribute__((target("avx2"),always_inline)) static inline void csa_add3(
    __m256i &hi,__m256i &lo,__m256i a,__m256i b,__m256i c) {
  __m256i u=_mm256_xor_si256(a,b);
  hi=_mm256_or_si256(_mm256_and_si256(a,b),_mm256_and_si256(u,c));
  lo=_mm256_xor_si256(u,c);
}
__attribute__((target("avx2"),always_inline)) static inline __m256i csa_lut8(__m256i x) {
  return _mm256_add_epi8(
    _mm256_shuffle_epi8(VLUT,_mm256_and_si256(x,VLOW4)),
    _mm256_shuffle_epi8(VLUT,_mm256_and_si256(_mm256_srli_epi16(x,4),VLOW4)));
}
__attribute__((target("avx2"),always_inline)) static inline __m256i csa_count(__m256i x) {
  __m256i v=_mm256_add_epi8(
    _mm256_shuffle_epi8(VLUT,_mm256_and_si256(x,VLOW4)),
    _mm256_shuffle_epi8(VLUT,_mm256_and_si256(_mm256_srli_epi16(x,4),VLOW4)));
  return _mm256_sad_epu8(v,_mm256_setzero_si256());
}

static void run_avx2(int n, int q, int *q_x, int *q_y, int *q_len, unsigned *ans) {
  (void)n;
  for (int gi = 0; gi < 512; gi++) {
    int g = (gi >> 2) & 63;
    int ph = gi & 3;                                  /* this sub-pass's phase class */
    if (GHEAD[gi] == GHEAD[gi + 1]) continue;
    const unsigned long long *S0, *S1;
    if ((gi >> 2) < 64) { S0 = F0; S1 = F1; } else { S0 = E0; S1 = E1; }
    int corr = 0;                                     /* B pointer correction (words) */
    if (g) {              /* rotation, built AT THIS PHASE: dst[m] = src_rot[m + ph] */
      unsigned long long sh = 64 - g;
      corr = ph;
      {                                                /* vectorised rotation: 4 words per step */
        __m128i vg = _mm_cvtsi32_si128((long long)g), vsh = _mm_cvtsi32_si128((long long)sh);
        int i = ph, lim = g_nw + 2;
        for (; i + 4 <= lim; i += 4) {
          __m256i a0 = ldu256(S0 + i);
          __m256i b0 = ldu256(S0 + i + 1);
          __m256i a1 = ldu256(S1 + i);
          __m256i b1 = ldu256(S1 + i + 1);
          _mm256_store_si256((__m256i *)(R0 + i - ph), _mm256_or_si256(_mm256_srl_epi64(a0, vg), _mm256_sll_epi64(b0, vsh)));
          _mm256_store_si256((__m256i *)(R1 + i - ph), _mm256_or_si256(_mm256_srl_epi64(a1, vg), _mm256_sll_epi64(b1, vsh)));
        }
        for (; i <= g_nw + 1; i++) {
          R0[i - ph] = (S0[i] >> g) | (S0[i + 1] << sh);
          R1[i - ph] = (S1[i] >> g) | (S1[i + 1] << sh);
        }
      }
      S0 = R0; S1 = R1;
    }
    for (int t = GHEAD[gi]; t < GHEAD[gi + 1]; t++) {
      unsigned long long rq = RQ[t];
      int k = ORD[t];
      int x = (int)(rq & 0x1FFFFFu), y = (int)((rq >> 21) & 0x1FFFFFu), l = (int)(rq >> 42);
      int delta = y - x;
      int dpos = delta >= 0;
      int dw = dpos ? (delta >> 6) : ((-delta) >> 6);
      /* branchless: the query start is min(x,y) and the A side is (E0,E1) when
         y>=x else (F0,F1).  The 50/50 branch here mispredicts on half the queries. */
      int s = dpos ? x : y;
      const unsigned long long *AB = dpos ? E0 : F0, *AC = dpos ? E1 : F1;
      int w0, wl, d, lo, hi;
      const unsigned long long *A0, *A1, *B0, *B1;
      w0 = s >> 6; wl = (s + l - 1) >> 6; d = wl - w0; lo = s & 63; hi = ((s + l - 1) & 63) + 1;
      A0 = AB + w0; A1 = A0 + AUXSTRIDE; B0 = S0 + w0 + dw - corr; B1 = B0 + AUXSTRIDE;
      int wA = w0 & ~3, wZ = wl & ~3;
      int o = w0 - wA, oL = wl - wZ;
      const unsigned long long *pA0 = A0 - o, *AA1 = A1 - o, *pB0 = B0 - o, *BB1 = B1 - o;
      int bend = wZ - wA;
      __m256i accv = _mm256_setzero_si256();
      __m256i bb = _mm256_setzero_si256();
      if (bend == 0) {
        __m256i z = _mm256_or_si256(
            _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)pA0), ldu256(pB0)),
            _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)AA1), ldu256(BB1)));
        z = _mm256_and_si256(z, _mm256_and_si256(FMASK[o * 64 + lo], LMASK[oL * 64 + hi - 1]));
        __m256i c8 = _mm256_add_epi8(
            _mm256_shuffle_epi8(VLUT, _mm256_and_si256(z, VLOW4)),
            _mm256_shuffle_epi8(VLUT, _mm256_and_si256(_mm256_srli_epi16(z, 4), VLOW4)));
        accv = _mm256_add_epi64(accv, _mm256_sad_epu8(c8, _mm256_setzero_si256()));
      } else {
        {
          __m256i z = _mm256_or_si256(
              _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)pA0), ldu256(pB0)),
              _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)AA1), ldu256(BB1)));
          z = _mm256_and_si256(z, FMASK[o * 64 + lo]);
          __m256i c8 = _mm256_add_epi8(
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(z, VLOW4)),
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(_mm256_srli_epi16(z, 4), VLOW4)));
          accv = _mm256_add_epi64(accv, _mm256_sad_epu8(c8, _mm256_setzero_si256()));
        }
        int b = 4; int cnt = 0;
        __m256i z=_mm256_setzero_si256(), a1=z, a2=z, a4=z, a8=z, t0=z, t1=z, t2=z, t3=z, t4=z, t5=z, t6=z, t7=z, t8=z, t9=z, t10=z, t11=z, t12=z, t13=z, fl=z;
        for (; b + 64 <= bend; b += 64) {
#define CD(K) csa_diff1(pA0,pB0,b+(K))
          csa_add3(t0,a1,a1,CD(0),CD(4));
          csa_add3(t1,a1,a1,CD(8),CD(12));
          csa_add3(t2,a2,a2,t0,t1);
          csa_add3(t3,a1,a1,CD(16),CD(20));
          csa_add3(t4,a1,a1,CD(24),CD(28));
          csa_add3(t5,a2,a2,t3,t4);
          csa_add3(t6,a4,a4,t2,t5);
          csa_add3(t7,a1,a1,CD(32),CD(36));
          csa_add3(t8,a1,a1,CD(40),CD(44));
          csa_add3(t9,a2,a2,t7,t8);
          csa_add3(t10,a1,a1,CD(48),CD(52));
          csa_add3(t11,a1,a1,CD(56),CD(60));
          csa_add3(t12,a2,a2,t10,t11);
          csa_add3(t13,a4,a4,t9,t12);
          csa_add3(fl,a8,a8,t6,t13);
          bb=_mm256_add_epi8(bb,csa_lut8(fl));
          if (__builtin_expect(++cnt == 16, 0)) { accv=_mm256_add_epi64(accv,_mm256_slli_epi64(_mm256_sad_epu8(bb,z),4)); bb=z; cnt=0; }
#undef CD
        }
        accv=_mm256_add_epi64(accv,_mm256_slli_epi64(_mm256_sad_epu8(bb,z),4));
        if (b > 4) {
          accv=_mm256_add_epi64(accv,csa_count(a1));
          accv=_mm256_add_epi64(accv,_mm256_slli_epi64(csa_count(a2),1));
          accv=_mm256_add_epi64(accv,_mm256_slli_epi64(csa_count(a4),2));
          accv=_mm256_add_epi64(accv,_mm256_slli_epi64(csa_count(a8),3));
        }
        for (; b < bend; b += 4) {
          __m256i z = csa_diff1(pA0,pB0,b);
          __m256i c8 = _mm256_add_epi8(
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(z, VLOW4)),
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(_mm256_srli_epi16(z, 4), VLOW4)));
          accv = _mm256_add_epi64(accv, _mm256_sad_epu8(c8, _mm256_setzero_si256()));
        }
        {
          __m256i z = _mm256_or_si256(
              _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)(pA0 + bend)), ldu256(pB0 + bend)),
              _mm256_xor_si256(_mm256_load_si256((const __m256i *)(const void *)(AA1 + bend)), ldu256(BB1 + bend)));
          z = _mm256_and_si256(z, LMASK[oL * 64 + hi - 1]);
          __m256i c8 = _mm256_add_epi8(
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(z, VLOW4)),
              _mm256_shuffle_epi8(VLUT, _mm256_and_si256(_mm256_srli_epi16(z, 4), VLOW4)));
          accv = _mm256_add_epi64(accv, _mm256_sad_epu8(c8, _mm256_setzero_si256()));
        }
      }
      { unsigned long long t[4]; _mm256_storeu_si256((__m256i *)t, accv);
        ans[k] = (unsigned)l - (unsigned)(t[0] + t[1] + t[2] + t[3]); }
    }
  }
}

#pragma GCC pop_options

static void run_scalar(int n, int q, int *q_x, int *q_y, int *q_len, unsigned *ans) {
  (void)n;
  for (int gi = 0; gi < 512; gi++) {
    int g = (gi >> 2) & 63;
    int ph = gi & 3;
    if (GHEAD[gi] == GHEAD[gi + 1]) continue;
    const unsigned long long *S0, *S1;
    if ((gi >> 2) < 64) { S0 = F0; S1 = F1; } else { S0 = E0; S1 = E1; }
    int corr = 0;
    if (g) {                                          /* build THIS sub-pass's rotation in place */
      unsigned long long sh = 64 - g;
      corr = ph;
      for (int i = ph; i <= g_nw + 1; i++) {
        R0[i - ph] = (S0[i] >> g) | (S0[i + 1] << sh);
        R1[i - ph] = (S1[i] >> g) | (S1[i + 1] << sh);
      }
      S0 = R0; S1 = R1;
    }
    for (int t = GHEAD[gi]; t < GHEAD[gi + 1]; t++) {
      unsigned long long rq = RQ[t];
      int k = ORD[t];
      int x = (int)(rq & 0x1FFFFFu), y = (int)((rq >> 21) & 0x1FFFFFu), l = (int)(rq >> 42);
      int delta = y - x;
      int dpos = delta >= 0;
      int dw = dpos ? (delta >> 6) : ((-delta) >> 6);
      /* branchless: the query start is min(x,y) and the A side is (E0,E1) when
         y>=x else (F0,F1).  The 50/50 branch here mispredicts on half the queries. */
      int s = dpos ? x : y;
      const unsigned long long *AB = dpos ? E0 : F0, *AC = dpos ? E1 : F1;
      int w0, wl, d, lo, hi;
      const unsigned long long *A0, *A1, *B0, *B1;
      w0 = s >> 6; wl = (s + l - 1) >> 6; d = wl - w0; lo = s & 63; hi = ((s + l - 1) & 63) + 1;
      A0 = AB + w0; A1 = A0 + AUXSTRIDE; B0 = S0 + w0 + dw - corr; B1 = B0 + AUXSTRIDE;
      unsigned dd = 0;
      for (int w = 0; w <= d; w++) {
        unsigned long long m = (A0[w] ^ B0[w]) | (A1[w] ^ B1[w]);
        if (w == 0) m &= ~WMASK[lo];
        if (w == d) m &= WMASK[hi];
        dd += (unsigned)__builtin_popcountll(m);
      }
      ans[k] = (unsigned)l - dd;
    }
  }
}

__attribute__((target("avx2,popcnt"))) static void build_all(int n, const char *s1, const char *s2) {
  int nw = (n + 63) >> 6;
  for (int i = 0; i <= nw + 1; i++) { E0[i] = E1[i] = F0[i] = F1[i] = 0; }
  {
    /* Vectorised plane build.  E0 = (a != '0'), E1 = (a == '2'),
       F0 = (c != '1'), F1 = (c == '0')   -- algebraically identical to the
       branchy version, but with no data-dependent branches at all
       (the scalar form mispredicts ~4x per 3 characters on random input). */
    const __m256i Z0 = _mm256_set1_epi8('0'), Z1 = _mm256_set1_epi8('1'), Z2 = _mm256_set1_epi8('2');
    int i = 0;
    for (; i + 32 <= n; i += 32) {
      int w = i >> 6, b = i & 63;
      __m256i v1 = _mm256_loadu_si256((const __m256i *)(s1 + i));
      __m256i v2 = _mm256_loadu_si256((const __m256i *)(s2 + i));
      unsigned e0 = ~(unsigned)_mm256_movemask_epi8(_mm256_cmpeq_epi8(v1, Z0));
      unsigned e1 = (unsigned)_mm256_movemask_epi8(_mm256_cmpeq_epi8(v1, Z2));
      unsigned f0 = ~(unsigned)_mm256_movemask_epi8(_mm256_cmpeq_epi8(v2, Z1));
      unsigned f1 = (unsigned)_mm256_movemask_epi8(_mm256_cmpeq_epi8(v2, Z0));
      if (b == 0) { E0[w] = e0; E1[w] = e1; F0[w] = f0; F1[w] = f1; }
      else { E0[w] |= (unsigned long long)e0 << 32; E1[w] |= (unsigned long long)e1 << 32;
             F0[w] |= (unsigned long long)f0 << 32; F1[w] |= (unsigned long long)f1 << 32; }
    }
    for (; i < n; i++) {
      int w = i >> 6, b = i & 63;
      int a = s1[i] - '0', c = s2[i] - '0';
      if (a >= 1) E0[w] |= 1ull << b;
      if (a >= 2) E1[w] |= 1ull << b;
      int g = (c + 2) % 3;
      if (g != 0) F0[w] |= 1ull << b;
      if (g >= 2) F1[w] |= 1ull << b;
    }
  }
  g_nw = nw;
  for (int i = 0; i < 64; i++) WMASK[i] = (1ull << i) - 1ull;
  WMASK[64] = ~0ull;
  for (int o = 0; o < 4; o++)
    for (int lo = 0; lo < 64; lo++) {
      unsigned long long *p = (unsigned long long *)&FMASK[o * 64 + lo];
      for (int j = 0; j < 4; j++) p[j] = (j < o) ? 0ull : (j == o ? ~WMASK[lo] : ~0ull);
    }
  for (int o = 0; o < 4; o++)
    for (int hh = 1; hh <= 64; hh++) {
      unsigned long long *p = (unsigned long long *)&LMASK[o * 64 + hh - 1];
      for (int j = 0; j < 4; j++) p[j] = (j < o) ? ~0ull : (j == o ? WMASK[hh] : 0ull);
    }
}

static inline int have_avx2(void) {
  unsigned a, b, c, d;
  __asm__ volatile("cpuid" : "=a"(a), "=b"(b), "=c"(c), "=d"(d) : "a"(1), "c"(0));
  int osxsave = (int)((c >> 27) & 1u), avx = (int)((c >> 28) & 1u);
  if (!(osxsave && avx)) return 0;
  unsigned lo, hi;
  __asm__ volatile("xgetbv" : "=a"(lo), "=d"(hi) : "c"(0));
  if ((lo & 6u) != 6u) return 0;
  __asm__ volatile("cpuid" : "=a"(a), "=b"(b), "=c"(c), "=d"(d) : "a"(7), "c"(0));
  return (int)((b >> 5) & 1u);
}

void solve(SOLVE_ARGS) {
  static void *ar = 0;
  if (!ar) { ar = arena_alloc(0); bind_arena(ar); }
  build_all(n, s1, s2);
  // group queries by the relative bit offset (sign aware), counting sort
  for (int i = 0; i <= 512; i++) GHEAD[i] = 0;
  for (int k = 0; k < q; k++) {
    int d = q_y[k] - q_x[k];
    int ad = (d >= 0 ? d : -d);
    int g = ((d >= 0) ? 0 : 64) + (ad & 63);
    GHEAD[g * 4 + ((ad >> 6) & 3) + 1]++;
  }
  for (int i = 0; i < 512; i++) GHEAD[i + 1] += GHEAD[i];
  for (int i = 0; i < 512; i++) GCUR[i] = GHEAD[i];
  for (int k = 0; k < q; k++) {
    int x = q_x[k], y = q_y[k], l = q_len[k];
    int d = y - x;
    int ad = (d >= 0 ? d : -d);
    int g = ((d >= 0) ? 0 : 64) + (ad & 63);
    int p = GCUR[g * 4 + ((ad >> 6) & 3)]++;
    ORD[p] = k;
    RQ[p] = (unsigned long long)(unsigned)x | ((unsigned long long)(unsigned)y << 21)
          | ((unsigned long long)(unsigned)l << 42);
  }

#ifdef TESTSCALAR
  run_scalar(n, q, q_x, q_y, q_len, ans);
#else
  if (have_avx2()) run_avx2(n, q, q_x, q_y, q_len, ans);
  else run_scalar(n, q, q_x, q_y, q_len, ans);
#endif
}

CompilationN/AN/ACompile ErrorScore: N/A


Judge Duck Online | 评测鸭在线
Server Time: 2026-11-09 04:45:37 | Loaded in 1 ms | Server Status
个人娱乐项目,仅供学习交流使用 | 捐赠