// References:
// [1] saffah_cc_v41_agg1, https://duck.ac/submission/107225: copied the accepted CDQ engine and its inherited attribution comments. No separate license notice appears.
// [2] saffah_codex_6s_agg2, https://duck.ac/submission/107718: directly reused the u16 prefix/bitmap fusion port to 1011a. No separate license notice appears.
// Approach: Preserve the exact 1011a CDQ computation; prefetch cross-dimensional gather values eight events ahead, use a 2304-event leaf threshold, and switch CT0 to contiguous clearing when the number of dirty entries exceeds CTN/64. These choices preserve outputs and reduce memory stalls or reset work.
// Purpose: Experimentally test whether the combined official speed reaches the current strict 1011a target.
// References:
// [1] saffah_cc_v41_agg1, https://duck.ac/submission/107225: copied the accepted 1011a CDQ engine and its inherited references; no separate license notice appears.
// [2] saffah_codex_6s_agg2, https://duck.ac/submission/107643 and work/1010a/fused_prefix_bmp.cpp: directly reused our u16 prefix/bitmap fusion.
// Approach: In each 1011a leaf, generate the bitmap of nontrivial buckets during the prefix scan of raw u16 counts, then scatter and sort exactly as in [1].
// Purpose: Experiment to reduce the official five-dimensional runtime and approach the dynamic threshold.
// ===== REFERENCES =====
// [1] duck.ac 用户 saffah_cc_v41_agg1(本账号),提交 **#107012**
// <https://duck.ac/submission/107012>(70.604696 ms = 我方当值最好件)
// —— **本文件正文的逐字节基座**(本地件 `problems/1011a/work/z11d_base.cpp`,md5 b34328f9acd16b90e0a2a6df886d7259)。
// [2] duck.ac 提交 **#107141**(用户 saffah_codex_6s_agg2,题目 **1011**)
// <https://duck.ac/submission/107141> —— 本发 (a)(b)(c) 三处的**来源**。
// ★ 取证(`§2.19.681/688`):该件与本账号在 1011 上的公开提交 **#107155 正文逐字节相同**
// (1011 的 notes `㉑` 记录:52348 B,diff = 0 行,md5 5f667c052ded16483c89d5affd06bd35)
// ⇒ **这不是"抄对手",是"把我们自己在姊妹题上已交付的刀搬回本题"** ✓
// 本发按 1011a 自己的文本重打,未改动其任何一行,且**只取 (a)(b)(c) 三处**(其 `SM7_SNET` 桶内排序网络
// 本席已在 1011a 上单变量实测为中性 +0.05%,故**不取**);其余一切差异(本文件保留 1011a 的
// `QK0/Q12/QIN` 查询侧打包、`AX_CROSSPF 4`、locality NTA、无 `rename-registers`)**一律不动**。
// [3] duck.ac 提交 **#106968**(用户 saffah_codex_6s_agg2,70.621149 ms = 本题当代 T)
// —— **只用于门槛计算**(严支 `0.99*T+1us = 69.915938`)。**本发未取用其任何一行。**
// [4] 本工作区 `BRIEF.md` **§2.19.652**(表示级/算法级普查)与协调者 `§2.19.690` 刀谱 ⑤(SIMD 前缀和 + u32→u16)。
// 许可:正文来自本账号 [1] 的公开提交与 [2](该件正文 = 本账号 #107155),无第三方版权内容。
// ======================
// ===== 思路 =====
// 【口径】`mine = 70.604696 (#107012)` · `T = 70.621149 (#106968)` · 严支 `0.99*T+1us = 69.915938`
// ⇒ 缺口 **0.688758 ms(0.976%)**。按 `§2.19.650` 这是 `0.99·T` 的手续费 ⇒ 必须真减工作量。
//
// 【第 ⓪ 步:诊断(判题机相位的本机代理 + 计数插桩,`work/z11d_phase_build.py`)】
// 本席先按 `§2.19.652` 做全部数组的值域普查(生产 n=1e5 实测最大,checksum 逐位同):
// IK/IC0/IC1/IC2 键与坐标 = 99999(17 位)/ QK0,Q12 半边 = 99999 / QIN 高半 id = 99999
// ⇒ **坐标一律 17 位 ⇒ u16 半宽刀在本题叶数组上无靶**(与同族 1011 的 19 位普查同结论)✓
// QIN 低半(叶内插入游标 ni)= 838、叶内 m/ni/nq = 1636/840/838(11 位)⇒ 只有游标能收窄,但**不按元素存储** ✓
// ⇒ 于是转向普查第二位:**叶的计数排序里被"桶数"而非"元素数"缩放的那部分**。
// 计数插桩(`work/z11d_phase_build.py`,生产 n=1e5)实测:
// 叶 `m` = 1,949,581 事件 / 992 次需要 out 的叶调用(均值 m ≈ 1560),而
// **桶数 `NB = LFB = ((n-1) >> (LFSH+LFD)) + 2 = 3126`** —— **桶数是元素数的 2 倍**
// ⇒ 清桶 3.1M 次、**标量前缀和 3.1M 次串行链**、**空桶走查 3.1M 次**,全都被 NB 缩放。
//
// 【本发这一刀:叶基数排序的表示级重写(三处,同源于 [2],取自其正文逐字)】
// (a) `u32 cnt[LEAFNB+2]` → **`u16 cnt[LEAFNB+2]`**
// **值域证明(可移植、非数据相关)**:`cnt[]` 的**唯一**写入点是 `cnt[b]=0`、
// `cnt[b+1]++`(= 一个桶的元素数)以及二者的前缀 ⇒ 任何活着的表项 ≤ 本叶事件数 `m`;
// 而这段代码**只在 `m <= KG_BASE = 2048` 时可达**(那正是进入 `base_case_t` 的守卫)
// ⇒ **恒 ≤ 2048 < 65536**,与数据分布无关 ✓
// (b) 标量前缀链 `for (b) cnt[b+1] += cnt[b];` → **`leaf_prefix16()`**(AVX2 16 lane 扫描)
// —— 3126 步串行 store-to-load 链 → 196 步向量;且 u16 让**每向量 lane 数 8 → 16 翻倍**
// (`§2.19.690` ①:前端/延迟绑时**只有"每向量 lane 数翻倍"有价**)✓
// (c) 全量桶走查 → **`[BMP]` 占用位图走查**:位图由**桶起点对**向量化构建(每 64 桶一个 u64 累加器,
// 无读-改-写链;`vpcmpgtw` + `vpmovmskb` + `pext`),然后只访问**置位**的桶
// ⇒ 走查从 NB=3126 次降到"49 个字 + 实际需要排序的桶数(≈333)" ✓
// **三处都不改"哪些元素进哪个桶"**:桶函数、前缀语义、桶内插入排序的比较键(完整 u64)一字未动
// ⇒ **输出是同一个有序数组** ⇒ 逐位相同是构造性的。
//
// 【闸门(全过)】
// * `work/m1c_vfy.py`(同一二进制内双引擎逐元素位比对)n=1000/10000/**100000** ⇒
// **bad=0,FNV 逐位相同**:eec697371a8b15de / e2663d10933814b4 / **8ecd81ec7fa3ea9e**(生产规模在列)✓
// * 独立 harness 下 `sum=309765069`、`fnv=0a33d8eaa7f40d10` 与基座**逐位相同** ✓
// * 判题机同款 **gcc 9.3 `-O2 -g -std=c++17` COMPILE OK** ✓;`-O2 -static -U_FORTIFY_SOURCE -std=c++17`
// (不加 -DLOCAL_TEST,判题机口径)0 error ✓
// * `[BMP]` 位图构建的**越界界**:`cnt` 4098 项,最大读下标 NB+16 = 3142 < 4098 ✓;`bmp` 66 项,NW=49 ✓
// 【本机 A/B(交错 min-of-4,仅参考;本机对预取/布局类读数不可信)】基座 269.4 M tick → 本件 260.0 M tick
// ================
// ===== REFERENCES =====
// [1] 本账号 saffah_cc_v41_agg1 提交 **#106755** <https://duck.ac/submission/106755>(**72.312243 ms**)
// —— 本发的**直接基座**:本文件正文 = 该件正文(含已发的查询侧打包刀)+ **唯一一处改动**(叶 `base_case_t` 查询块循环入口各加
// 1 条对结算目标的前瞻预取,见思路区)。
// [2] 本账号 saffah_cc_v41_agg1 提交 **#106675** <https://duck.ac/submission/106675>(题目 **1013**,5232.965810 ms)
// —— **同形态先例**:在 1013 上给"**结算型随机 RMW**"(`out[i] += add`,地址来自**顺序**的 `s_qord[]`)
// 加前瞻预取,一手 **−126.497 ms**(该题最大一刀)⇒ 配方前提 = "前瞻地址来自顺序数组 ⇒ 零延迟可算"。
// [3] 本席判题机探针 `problems/1011a/work/b13w_cmpacc.cpp`(`tools/probe.py`,Run Finished;
// **arm0 checksum = 314266751 = 家族常数** ⇒ 正例对照 ✓,两轮交错):
// arm0 基准 265.72 / 264.73 Mcyc
// arm1 **去掉 `OUT[QID[q]] += …` 这条 RMW 存** ⇒ **−6.8~7.9 Mcyc = 2.6~3.0% 全时**(本发靶池)
// arm2 去掉整条比较链 ⇒ −71.4~72.4 Mcyc = 27.0~27.3% 全时(**该格判定"接近结构地板、不出刀"**:
// 4 维**合取**支配判定不能压成一条 64 位比较,且已 4/3/2 宽 SIMD 密排)
// ======================
// [5] 本账号 saffah_cc_v41_agg1 提交 **#106797**(73.127612 ms)—— 同一处预取**先发在旧基座 #106404 上**(未含打包刀),
// 读数为 73.127612 vs 旧基座 73.401858 = **−0.374 ms(−0.51%)** ⇒ 该预取单独为正 ✓;
// 本发把它**搬到当代最好件 #106755 的正文上**(同一处改动、同样的单变量对照:vs #106755 只差这 3 条预取)✓
// ======================
// ===== 思路 =====
// 靶:叶 `base_case_t` 里**每个查询一次**的结算存 `OUT[QID[q]] += hsum256(...)`(判题机实测池 = **2.6~3.0% 全时**)。
// 形态与 `#106675`(1013 最大一刀)同族;本处**配方前提成立**:该 RMW 的地址 `OUT[QID[q + D]]` 的**下标来自顺序数组 `QID[]`**
// (在叶内按查询顺序填),因此**领先 D 个查询的目标行地址可在零延迟内算出**,不需要第二级依赖 ✓
// 本发唯一改动:在三个查询块循环(4/3/2 宽)入口各加 **1 条**
// `if (q + AX_OUTPF < nq) __builtin_prefetch(&OUT[QID[q + AX_OUTPF]], 1, 3);`
// `AX_OUTPF = 32`(两个旋钮值以外无任何改动;单变量、距离类旋钮逐点真发 ✓)。
// 正确性:**只新增预取、不改变任何读写值**(`QID[...]` 的读无副作用)⇒ 逐位相同是构造性的 ✓
// 闸门:`work/m1c_vfy.py`(双引擎同进程逐元素)n=1000/10000/100000 ⇒ **全部 bad=0、哈希逐位相同**
// (`eec697371a8b15de` / `e2663d10933814b4` / `8ecd81ec7fa3ea9e`,与基座一致 ✓);
// 判题机口径 `-O2 -static -U_FORTIFY_SOURCE -std=c++17` 编译 0 error ✓
// 预期:按 1013 同形态刀的兑现率(池的 ~1/3~1/2)⇒ **−0.9~1.5% 全时**(真靶差 1.03%)。
// ======================
// ===== 正文 =====
// ===== REFERENCES =====
// [1] 本账号 saffah_cc_v41_agg1 提交 #106229 <https://duck.ac/submission/106229>
// —— 本发的**直接基底**(74.922356 ms,1011a 当前最好件)。除「思路」写明的那一处改动外,
// 本文件与它**逐字节相同**。
// [2] 本账号 saffah_cc_v41_agg1 提交 #106360 <https://duck.ac/submission/106360>
// —— **同族姊妹题 1011e6**(同一支 CDQ 引擎的 k=5 实例,n=1e6)上已交付并判题机实测 −1.85% 的一发,
// 文件 `problems/1011e6/work/sm7_xpf8_sub.cpp`。本发从它**逐字移植两件事**:
// (a) `merge_loops` 的**指针游标循环形态**(`pa/pb/po/pc`,不再有 `ia/ib/t` 索引;
// `keep` 合成一条 `1 - (takeA ^ (v>>42))`);
// (b) `MKE_CROSS` 的 embed gather 的**前瞻预取**(该文件 `SG_CROSSPF=8` 那两行)。
// 两者的出处都是本家族这一支 CDQ 引擎(链路见 `problems/1011e6/notes.md` 14:0x 一节)。
// [3] 判题机 gcc 9.3 真汇编(仪器 `work/a1x_g93.py`,缓存 `work/a1x_g93/a1x_base.s`):
// `solve_b<5,2,0>` 主归并环 = **27 条指令/事件**,其中掩码 `movabs r15, 4398048608255`
// **在环内**、`cmp r15, rdx` 在环内**出现两次**(一次 `setb`、一次 `cmovbe`)、
// 索引符号扩展 `movsx` **3 条**(`ia`、`ib`、`cx`)。
// [4] 判题机 2×2 臂矩阵(**本发的定价依据**):`work/a1x_probe_merge.py` -> `work/a1x_probe_merge.cpp`,
// `tools/probe.py C++17`,n=1e5 k=5,四臂交错 ×2 轮,四臂 out[] checksum **全部 314266751**:
// arm0 stock 268.2 / arm1 指针 271.0 (+1.04%) / arm2 stock+PF8 267.7 (−0.19%) /
// **arm3 指针+PF8 262.3 (−2.20%)**。
// =======================
// ===== 思路 =====
// **本发 = 姊妹题 1011e6 上已兑现的"归并形态 + gather 预取"两手,按判题机 2×2 矩阵整格移植。**
//
// 【为什么打这里:本地插桩相位账】(`work/a1x_phase_build.py` -> `work/a1x_phase.cpp`,rdtscp,
// n=1e5 k=5,checksum 314266751 与家族常数一致;三轮 min 口径):
// | 段 | 占全时 | 规模 |
// |---|---|---|
// | **merge(含 cross 构建)** | **24.5~27.3%** | 5,250,479 事件 / cx 合计 2,627,010 |
// | leaf(base_case_t:事件构建+排序+比较环) | **61.9~64.8%** | 1,949,441 事件 / 1248 次调用 |
// | ctr_solve | 7.9~8.5% | — |
// | setup + 顶层发射 | 2.1~2.9% | — |
// merge = 5.25M 事件 × ~27 uop ⇒ 实测 ~10~12 cyc/事件(≈2.3~2.7 uop/cyc)⇒ **前端/uop 受限**;
// 而缺口只有 **1.006%**(753.853 µs / 74.922356 ms)⇒ merge 掉 ≥4% 即够。
//
// 【为什么是"整格"而不是两手分开交】**单变量纪律的例外**:本格是**已定价**的单一体
// (军规见 BRIEF §2.19 的"定价"一节:**测过的组合可整格交**)——因为
// **指针形态单独交是 +1.04%、预取单独交是 −0.19%(都在噪声内或更差),只有整格是 −2.20%**。
// 机理:指针形态把 3 条索引符号扩展(`movsx`)从环里拿掉、腾出 GPR ⇒ 预取的两条地址计算
// (`pa[8]` 载入 + `eid()` 掩码 + LEA)**才放得进环**;索引形态里预取的额外 `movsx` 把收益吃掉。
// ⇒ 这正是"**换别处就得重新定价**"的又一实例(§2.18.863②),也说明**本机构建不可作数**:
// 本机 g++15 上指针形态 +4.7%、指针+预取 −2.4%(符号与判题机 gcc 9.3 不同,§2.19.007 类)。
//
// 【预取为什么在本题仍然值钱】1011e6 上 `XEMB` 是 4 MB(L3/DRAM 随机行);本题 n=1e5 ⇒
// `XE` = `XD[3]`/`XD[4]` 各 **400 KB**。本发的 2×2 实测(arm2/arm3)说明:**随机行延迟仍未被完全
// 隐藏**(只是单靠预取、在索引形态下拿不到),⇒ "代价结构不同"是真的,但方向一致(§2.19.124 的
// "先证明结构在本题存在":结构存在 = `MKE_CROSS` 的 `XE[eid(V)]` 随机 gather,每 merge 事件一次)。
//
// 【闸门】①`work/m1c_vfy.py`(双引擎逐位)n=1000/10000/**100000** 输出全等 ✓;
// ②`zf_vfy.py` k=5 六组数据(n=1000/3000/12000/40000/40001/5)× O(n²) 暴力 **逐元素 FNV 全等** ✓;
// ③`-O2 -static -U_FORTIFY_SOURCE -std=c++17`(判题机口径,不加 `-DLOCAL_TEST`)0 error ✓。
// =======================
// ===== REFERENCES =====
// [1] 本账号 saffah_cc_v41_agg1 提交 #106164 <https://duck.ac/submission/106164>
// —— 本发的**直接基底**(75.151936 ms):把家族当前 1010a 引擎(#103203 正文)实例化到 5D 的件。
// [2] duck.ac 用户 saffah_codex_6s_agg2 提交 #106138 <https://duck.ac/submission/106138>
// —— 本轮把 1011a 打红的那一发(74.916670 ms)。其文件头自述:由他自己的 1010a 引擎 #106122
// 实例化到 5D;而 #106122 的 [0] 号引用正是**本账号的 #103203**。
// **取证**:把两边**正文**逐行 diff,全文只差 **3 处**(`problems/1011a/work/b3x_port_ev.py`):
// (a) 顶层首坐标分组发射环(我方两趟 + 随机 `X0[g_ord[..]]` 边界探测;他单趟 + `g_x[]` 顺序边界)
// —— 他件头自述出处是他自己的 #106067 / #106114;
// (b) `base_case_t` 三/四路叶里两处 `QN` 单调性 break(他自述出处 #103873:QN 是随扫描单调递增的
// 运行插入计数,该 break 永不触发;即使触发也只是把余下查询交给标量路径 ⇒ 同答案);
// (c) `KG_BASE` 3072(我方)vs 2048(他)。
// ⇒ 本发把 (a)(b) 原样移植,并采 (c) 的 2048(判题机同进程 probe 实测:新发射环下 2048 比 3072 快 0.46%)。
// [3] 本账号 #99923 <https://duck.ac/submission/99923> —— 上一代引擎(本次整代替换前的 1011a 最好件)。
// [4] Judge Duck FAQ(-O2 -static);定价仪器见下。
// [5] 本发 = [1] + **单变量** `KG_ARENA (128u << 20) → (16u << 20)`(事件竞技场声明尺寸 1 GB → 128 MB)。
// 动机:同为 CDQ 引擎的 1010 上,对手用的是 **16 MB**(`problems/1010/notes.md` 的取证清单 #5),
// 而本工作区从未给这一轴定价过;它是**布局漂移类**(改的是静态区尺寸 ⇒ 热数组的相对地址/组相联映射变)
// ⇒ 只能靠正式提交定价。安全性:n=1e5 时实测 checksum 与 128 MB 版**逐位相同**(314266751)⇒ 不溢出。
// =======================
// ===== 思路 =====
// **基座对齐**:把对手 #106138 相对我方 #106164 的**全部**三处正文差异逐行移植(`b3x_port_ev.py` 带 anchor
// 计数断言),并按判题机实测取回 `KG_BASE = 2048`。移植后本文件正文与 #106138 **只差注释**。
//
// 【判题机同进程 A/B(`zf_ab.py` 生成、`tools/probe.py` 立即体验,n=1e5 k=5,两臂 out[] 逐元素 FNV 相同)】
// | 臂 | 读数 | 说明 |
// |---|---|---|
// | 我方 #106164(旧环 + KG_BASE 3072) | 269.745 Mcyc / 74.93 ms | 真发 75.152 ms(比例 1.0030)|
// | 对手 #106138(新环 + 2048) | 267.551 Mcyc / 74.32 ms | 真发 74.917 ms |
// | 新环 + KG_BASE 3072 | 269.008 Mcyc | 移植 (a)(b) 的收益 = 0.27% |
// | **新环 + KG_BASE 2048(本发)** | **267.763 Mcyc / 74.38 ms** | **再 −0.46%**,与对手件等速 |
// ★ 结论:`KG_BASE` 的**最优值随发射环形态移动**(旧环上 3072 更优、新环上 2048 更优)
// —— 一个"换了别处就得重扫常量"的实例(§2.18.863②)。
//
// 【闸门】k=5 六组数据(n=1000/3000/12000/40000/40001/5;均匀/3 值/全 0/严格降序/严格升序/二值)
// `zf_vfy.py` 与 O(n²) 暴力**逐元素 FNV 全等** ✓;`-O2 -static -U_FORTIFY_SOURCE -std=c++17`
// (判题机口径,不加 LOCAL_TEST)0 error ✓。
// ================
// ===== REFERENCES =====
// [1] 本账号(saffah_cc_v41_agg1)提交 #103203 <https://duck.ac/submission/103203>
// —— **本文件正文的出处**:该发在姐妹题 1010a(k=4, n=1e5)上交付的
// 「维度特化 CDQ + 后缀型融合计数器([SFX])」引擎,42.297013 ms。
// 本文件 = 它的**逐字节正文**(`problems/1010a/work/a1r/sub10as.cpp`),**只换 k=5 入口**。
// [2] duck.ac 用户 saffah_codex_6s_agg2 提交 #106138 <https://duck.ac/submission/106138>
// —— 本轮把 1011a 打红的那一发(74.916670 ms)。取证结论(其文件头逐字自述):
// 它是把**他自己的 1010a 引擎**(#106122,而 #106122 的 [0] 号引用正是**我们的 #103203**)
// "instantiate ... for five dimensions" 得到的 5D 实例。⇒ **他赢在"把我们自己的 1010a 引擎
// 搬到 5D",而我们此前从未搬过**。
// [3] 本账号提交 #99923 <https://duck.ac/submission/99923> —— 本发之前在 1011a 上的最好件
// (79.185341 ms),是**上一代** CDQ 引擎(成对查询叶 + 3/4 路阶梯核 + LFB)。本发把它整代换掉。
// [4] 同族参数与闸门口径:`problems/1010a/notes.md` 第七轮([SFX] 的推导与判题机 −4.08%)·
// `problems/1010/notes.md`。构建:Judge Duck FAQ(-O2 -static)。
// =======================
// ===== 思路 =====
// **单变量:换引擎代际**(`#99923` 的旧 CDQ 引擎 ⇒ 家族当前 1010a 引擎),k=5 实例化。
//
// 【为什么这么打】取证(§2.18.347 一次命中):对手 #106138 的跳幅(78.713 → 74.917,−4.8%)
// **不是新刀,而是"把 1010a 的引擎实例化到 5D"** —— 而那份 1010a 引擎的祖先是**我们的 #103203**。
// ⇒ 本轮的正确动作是**基座对齐 + 取回自家引擎**(与 1013b 上"对手抄的是我们 1012b 的刀"同形,
// 只是方向反过来:这次是我们没把自家姐妹题的引擎搬过去)。
//
// 【本文件的 5D 实例化是**已存在**的模板路径】文件里 `count_5d` → `kg::run_t<5>` 本来就在
// (1010a/1011a 共用一份正文);本发**不加任何新代码**,只是把它交到 1011a 上。
//
// 【正确性闸门】(本发在 k=5 上是**首次**启用该路径,故按全闸门走)
// * `zf_vfy.py` 6 组数据(n=1000/3000/12000/40000/40001/5;均匀/3 值/全 0/严格降序/严格升序/二值)
// **逐元素 FNV-1a 与 O(n²) 暴力全等** ✓;
// * 同进程双变体 A/B(`zf_ab.py`,n=1e5 k=5,两臂 out[] 逐元素哈希**相同** = 314266751,
// 与 #99923 那条 lane 的哈希一致)⇒ 不是"测了个错的程序";
// * `-O2 -static -U_FORTIFY_SOURCE -std=c++17`(**不加** -DLOCAL_TEST,判题机口径)0 error。
//
// 【本地同进程 A/B(仅定性,判题机提交为准)】n=1e5 k=5,min-of-3 交替:
// 本文件 210.35 Mcyc | 对手 #106138 213.01 Mcyc ⇒ **−1.25%**(两臂 checksum 相同)。
// ⚠ 本地绝对值不可用(本族已多次印证);本发的定性依据是"同进程、同数据、同一次运行"。
// ================
// ===== REFERENCES =====
// [1] duck.ac 用户 saffah_codex_6s_agg2,提交 #103069 <https://duck.ac/submission/103069>
// 用途:**直接复制了该提交的全部正文作为基座**(本发的全部改动都在它之上,见「思路」)。
// [2] duck.ac 用户 saffah_codex_6s_agg2,提交 #103094 <https://duck.ac/submission/103094>
// 用途:**参考了它的"CT5 后缀增量"改写思想**(CT5INC 表 + `CT5[0..7] += CT5INC[v>>17]`,
// 查询退化成一条标量载入)。本发把同一条恒等变形推广到 CT1..CT4。
// ⚠ 本题(n=1e5 < 2^17)没有 CT5 这一层 —— [1] 已把它整层删除,故本发只后缀化 CT1..CT4。
// [3] duck.ac 用户 saffah_cc_v41_agg1(本账号),提交 #103161 <https://duck.ac/submission/103161>
// 用途:本账号上一发最好件 = #103069 的逐字节复刻(44.098440 ms)。
// [4] duck.ac 用户 saffah_cc_v41_agg1(本账号),提交 #101855 <https://duck.ac/submission/101855>
// 用途:本条 lane(维度特化 CDQ + 链式嵌入坐标 + 向量化叶 + 末维融合计数器)的来源。
// [5] duck.ac 用户 saffah_cc_v41_260924,提交 #101795 <https://duck.ac/submission/101795>
// 用途:模板维度分发的早期形态(由 [4] 的头部注释指出)。
// ======================
// ===== 思路 =====
// 唯一改动 = **[SFX] 计数器后缀化**(`grep -n 'SFX' 本文件` 可见全部改动点;其余逐字节 = [1])。
//
// 末维融合计数器原本每一层都是「块内计数 + 查询时做一次掩码前缀归约」:
// CT_k[p] = 「块 p 内的插入数」;查询 = Σ_{p ∈ 本组, p < 目标} CT_k[p](4~10 条 uop 的向量归约)。
// 把同一层改成**后缀数组**([2] 在顶层 CT5 上已经证明可行的那个恒等变形):
// CT_k[p] := 「本组内块号 < p 的插入数」
// * 查询贡献 = 单个标量载入 `CT_k[目标]`(一次归约都不做);
// * 插入 = 对本组的 8 条车道做一次 `+= INC_k[块内序号]`,其中 `INC_k[r]` 的第 j 位 = (j > r)
// —— 一条 16/32 字节的 load-add-store。
// 各层贡献互不重叠、合起来覆盖 [0, v) 的完整区间,所以答案是**逐位恒等变形**。
//
// 精度(唯一约束,见头部注释「u16 block count must stay <= 32767」):
// 组内后缀值上界 = 8 × 本层块跨度 × 127(末维坐标重数上界)。
// CT1(跨度 32)= 32512 ✓ 仍是 u16;**CT2(跨度 256)= 260096 ✗ ⇒ 本发把 CT2 从 u16 加宽到 u32**
// (2 KB → 8 KB,全在 L1);CT3/CT4 本来就是 u32 ✓。
// CT0 无法后缀化(8×32×127 = 32512 > 255)⇒ 保留原来的 su8_32 归约。
//
// 判题机实测(立即体验,同进程交替臂序 A/B,工具 problems/1010/work/a1r/a1r_ab.py,
// n=100000 k=4,两臂整份 out[] 的 FNV-1a 逐元素哈希相同 ⇒ 不是"测了个错的程序"):
// 基座 #103069 157.525M cyc → +SFX 150.914M cyc = **−4.20%**。
// 正确性:`zf_vfy.py` 6 组数据(n=1000..40001,均匀/3值/全0/降序/升序/二值)与 O(n²) 暴力
// **逐元素 FNV-1a 全等**;另加 6 组更大规模(n=5000..100000)同样 v0==v1。
// **阳性对照**:把 CT1 的 INC 行号故意错一位 ⇒ n=40000/40001/5000/50000 立刻失配
// ⇒ 闸门确实压到了 ctr_solve 这条路径(不是空测)。
// ================
#ifndef MF_CTRPF
#define MF_CTRPF 0
#endif
#ifndef MF_CTR5
#define MF_CTR5 1
#endif
#ifndef MF_LEAFLP
#define MF_LEAFLP 32
#endif
#ifndef MF_LEAFG
#define MF_LEAFG 4
#endif
#pragma GCC optimize("O3","unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,popcnt,lzcnt")
// duck.ac 1010 (4D) / 1011 (5D) -- "count for every point the number of points that are
// smaller in *every* dimension" (n = 300000, coords in [0,n), strict comparisons).
//
// Algorithm: D&C over dimensions on an event sequence, keys carried inline.
// * 2n events: one query event + one insert event per point, keyed by dim 0
// (a query event is stored *before* an insert event of the same key, which encodes
// the strict-inequality rule).
// * solve(seq,m,d): seq is ordered by dim d and every event carries the coordinate of
// dim d+1 in bits 32..63 of its u64. Split by position, recurse both halves, then
// merge the two halves by the carried key (a plain unsigned `<` on the packed u64,
// so equal keys keep query-before-insert order); the same pass collects the cross
// events (left inserts + right queries), re-keys them to dim d+2 and recurses one
// dimension deeper. Ordering by dim d with that tie rule makes "insert before query"
// exactly "all dims so far strictly smaller", and equal coordinates are never counted.
// * d == K-1: every insert-before-query pair now qualifies, so one O(m) running counter
// finishes the innermost dimension (no log factor).
// * m <= 32: direct O(m^2) counting plus an insertion sort.
//
// Local timings (i5-6500 @ 3.2GHz, best of 3, n = 300000 uniform random):
// 4D 1.57 s 5D 3.85 s
// (Board #1: 508.596 ms / 745.358 ms -- this is still 3x / 5x behind.)
// k-dimensional dominance counting: out[i] = #{j : all dims of j < dims of i}
//
// Event-sequence CDQ with keys carried inline.
// event (u64) = (key << 32) | (ins << 31) | id ins = 1 for an "insert" event
// key = coordinate of dim (level+1): the field used by this level's merge.
// The packing makes a plain unsigned comparison put, for equal keys, the query
// event first -- exactly the strict-inequality rule (equal coords must not count).
//
// solve(seq, m, d): seq is ordered by dim d (tie rule above), elements carry the
// key for dim d+1; count pairs (insert before query) with dims d+1..K-1 strictly
// smaller; return `out` = the same events merged by the carried key (= dim d+1).
// Cross = A-inserts + B-queries re-keyed to dim d+2, recursed at level d+1.
// d == K-1 : every (insert before query) pair qualifies -> running counter, O(m).
#include <cstring>
#include <immintrin.h>
#pragma GCC push_options
#pragma GCC target("avx2,popcnt,bmi2")
#include <cstdint>
typedef unsigned u32;
typedef uint64_t u64;
typedef unsigned char u8;
typedef unsigned short u16;
#ifndef KG_COND
#define KG_COND 0
#endif
#ifndef KG_PF
#define KG_PF 0
#endif
#ifndef KG_BASE
#define KG_BASE 2304
#endif
#ifndef LEAFNB
#define LEAFNB 4096
#endif
#define KG_MAXK 5
#define KG_ARENA (16u << 20)
namespace kg {
// ------------ fused innermost-dimension counter (prefix count over [0,n)) -------------
// AVX2. Levels: u8 leaf /32 values, u16 per 32 and per 256, u32 per 2048/16384/131072.
// Every u16 block count must stay <= 32767 (madd uses signed lanes); the guard (max
// multiplicity of a last-dim coordinate <= 127) bounds 32*127=4064 and 256*127=32512.
#define CT_MAXV 1048580u
static u8 CT0[CT_MAXV];
static u16 CT1[CT_MAXV / 32 + 16] __attribute__((aligned(32)));
static u32 CT2[CT_MAXV / 256 + 16] __attribute__((aligned(32))); // [SFX] u16 -> u32
static u32 CT3[CT_MAXV / 2048 + 16] __attribute__((aligned(32)));
static u32 CT4[CT_MAXV / 16384 + 16] __attribute__((aligned(32)));
static u32 CT5[CT_MAXV / 131072 + 8];
// [SFX] suffix-counter increment masks. Row r holds a 1 in every lane strictly
// above r, so an insert landing in block r adds one to each suffix position of
// r's 8-block group that must start counting it.
static const u16 INC16[8][8] __attribute__((aligned(32))) = {
{0,1,1,1,1,1,1,1},{0,0,1,1,1,1,1,1},{0,0,0,1,1,1,1,1},{0,0,0,0,1,1,1,1},
{0,0,0,0,0,1,1,1},{0,0,0,0,0,0,1,1},{0,0,0,0,0,0,0,1},{0,0,0,0,0,0,0,0}};
static const u32 INC32[8][8] __attribute__((aligned(32))) = {
{0,1,1,1,1,1,1,1},{0,0,1,1,1,1,1,1},{0,0,0,1,1,1,1,1},{0,0,0,0,1,1,1,1},
{0,0,0,0,0,1,1,1},{0,0,0,0,0,0,1,1},{0,0,0,0,0,0,0,1},{0,0,0,0,0,0,0,0}};
#define SFX16(p, r) do { __m128i _a = _mm_load_si128((const __m128i *)(p)); \
__m128i _b = _mm_load_si128((const __m128i *)(INC16 + (r))); \
_mm_store_si128((__m128i *)(p), _mm_add_epi16(_a, _b)); } while (0)
#define SFX32(p, r) do { __m256i _a = _mm256_load_si256((const __m256i *)(p)); \
__m256i _b = _mm256_load_si256((const __m256i *)(INC32 + (r))); \
_mm256_store_si256((__m256i *)(p), _mm256_add_epi32(_a, _b)); } while (0)
static int FUSE_OK;
static inline u32 su8_32(const u8 *p, int cnt) {
__m256i v = _mm256_loadu_si256((const __m256i *)p);
__m256i m = _mm256_cmpgt_epi8(_mm256_set1_epi8((char)cnt),
_mm256_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31));
__m256i x = _mm256_and_si256(v, m);
__m256i s = _mm256_sad_epu8(x, _mm256_setzero_si256());
__m128i t = _mm_add_epi64(_mm256_castsi256_si128(s), _mm256_extracti128_si256(s, 1));
return (u32)((unsigned long long)_mm_cvtsi128_si64(t) + (unsigned long long)_mm_extract_epi64(t, 1));
}
static inline u32 su16_8(const u16 *p, int cnt) {
__m128i v = _mm_loadu_si128((const __m128i *)p);
__m128i x = _mm_and_si128(v, _mm_cmpgt_epi16(_mm_set1_epi16((short)cnt),
_mm_setr_epi16(0, 1, 2, 3, 4, 5, 6, 7)));
__m128i s = _mm_madd_epi16(x, _mm_set1_epi16(1));
s = _mm_add_epi32(s, _mm_shuffle_epi32(s, _MM_SHUFFLE(1, 0, 3, 2)));
s = _mm_add_epi32(s, _mm_shuffle_epi32(s, _MM_SHUFFLE(2, 3, 0, 1)));
return (u32)_mm_cvtsi128_si32(s);
}
static inline u32 su32_8(const u32 *p, int cnt) {
__m256i v = _mm256_loadu_si256((const __m256i *)p);
__m256i x = _mm256_and_si256(v, _mm256_cmpgt_epi32(_mm256_set1_epi32(cnt),
_mm256_setr_epi32(0, 1, 2, 3, 4, 5, 6, 7)));
__m128i t = _mm_add_epi32(_mm256_castsi256_si128(x), _mm256_extracti128_si256(x, 1));
t = _mm_add_epi32(t, _mm_shuffle_epi32(t, _MM_SHUFFLE(1, 0, 3, 2)));
t = _mm_add_epi32(t, _mm_shuffle_epi32(t, _MM_SHUFFLE(2, 3, 0, 1)));
return (u32)_mm_cvtsi128_si32(t);
}
static inline void ctr_ins(u32 v) {
CT0[v]++;
SFX16(CT1 + ((v >> 8) << 3), (v >> 5) & 7);
SFX32(CT2 + ((v >> 11) << 3), (v >> 8) & 7);
SFX32(CT3 + ((v >> 14) << 3), (v >> 11) & 7);
SFX32(CT4 + ((v >> 17) << 3), (v >> 14) & 7); // [SFX] 4 wide adds replace 4 RMWs
}
static inline void ctr_clr(u32 v) {
CT0[v]--; // [CTRR] CT1..CT5 are restored in bulk by ctr_reset_hi()
}
// [CTRR] live bound of the coordinate values seen by ctr_* (set in run_t)
static u32 CTN;
static inline void ctr_reset_hi(void) {
memset(CT1, 0, sizeof(u16) * (((size_t)CTN >> 5) + 2));
memset(CT2, 0, sizeof(u32) * (((size_t)CTN >> 8) + 2));
memset(CT3, 0, sizeof(u32) * (((size_t)CTN >> 11) + 2));
memset(CT4, 0, sizeof(u32) * (((size_t)CTN >> 14) + 2));
// CT5 is unnecessary for this problem: n = 100000 < 2^17.
}
static inline u32 ctr_qry(u32 v) {
u32 s = 0;
s += CT4[v >> 14]; // [SFX] scalar suffix load replaces the vector reduction
s += CT3[v >> 11];
s += CT2[v >> 8];
s += CT1[v >> 5];
s += su8_32(CT0 + (v & ~31u), (int)(v & 31));
return s;
}
#ifndef LFAUTO
#define LFAUTO 0
#endif
static int LFSH;
static int LFD;
static int LFB; // [v4] dynamic bucket count
static const u32 *XD[KG_MAXK];
static u32 *OUT;
static u64 *AB, *ABtop;
static inline int eid(u64 v) { return (int)(v & 0x1FFFFFu); }
static inline int lfbucket(u64 v) { return (int)(((u32)(v >> 43)) >> LFSH); }
static inline int eq_(u64 v) { return (int)((v >> 42) & 1); }
static inline int keep_test(int takeA, int role) { return 1 - (takeA ^ role); }
static inline u32 embnk(u64 v) { return (u32)((v >> 21) & 0x1FFFFFu); }
static inline u64 mk(u32 key, int q, int id, u32 nk) {
return ((u64)key << 43) | ((u64)(1 - q) << 42) | ((u64)nk << 21) | (u64)id;
}
// ---- compile-time depth/dimension specialisation (the fleet's `DN` recipe) ----
// Everything that used to be a runtime `if (d ...)` / `if (K ...)` inside the leaf and in
// the merge becomes a template constant, so the leaf kernels lose their per-iteration
// `nd > 0 / > 1 / > 2` tests AND are instantiated per remaining-dimension count.
template <bool B> struct TB {};
template <int V> struct NCL { static const int v = (V < 0) ? 0 : ((V > 3) ? 3 : V); };
template <int KT, int D, int NO> static inline void solve_t(u64 *seq, int m, u64 *out);
// resolve a fusion-level cross: events carry the last dim (K-1) coordinate in bits 32..63
// and are ordered by dim K-2; a left-insert before a right-query with a strictly smaller
// last-dim value is a hit.
static void ctr_solve(u64 *cross, int cx, u32 *dirt) {
int nd = 0;
for (int i = 0; i < cx; i++) {
#if MF_CTRPF > 0
if (i + MF_CTRPF < cx) {
u32 vf = (u32)(cross[i + MF_CTRPF] >> 43);
__builtin_prefetch(&CT0[vf], 1, 0);
__builtin_prefetch(&CT1[vf >> 5], 1, 0);
__builtin_prefetch(&CT2[vf >> 8], 1, 0);
}
#endif
u64 v = cross[i];
u32 val = (u32)(v >> 43);
if ((v >> 42) & 1) { ctr_ins(val); dirt[nd++] = val; }
else OUT[eid(v)] += ctr_qry(val);
}
if ((u32)nd > (CTN >> 6)) {
memset(CT0, 0, (size_t)CTN);
} else {
for (int i = 0; i < nd; i++) ctr_clr(dirt[i]);
}
ctr_reset_hi();
}
// small base: compact insert list + SIMD comparison, then insertion sort by the key
static const u32 VMASK[8][8] = {
{0,0,0,0,0,0,0,0},
{~0u,0,0,0,0,0,0,0},
{~0u,~0u,0,0,0,0,0,0},
{~0u,~0u,~0u,0,0,0,0,0},
{~0u,~0u,~0u,~0u,0,0,0,0},
{~0u,~0u,~0u,~0u,~0u,0,0,0},
{~0u,~0u,~0u,~0u,~0u,~0u,0,0},
{~0u,~0u,~0u,~0u,~0u,~0u,~0u,0}};
// [z11d] 16-lane u16 inclusive prefix sum, ported from the sibling 1011 engine
// (`leaf_prefix16`, problems/1011/work/z11c_sub1.cpp). Same output as the scalar
// `for (b) cnt[b+1] += cnt[b];` chain: lane-serial shifts within the ymm plus a
// per-iteration carry. u16 is safe because cnt[] only ever holds prefix counts
// bounded by the leaf's own m <= KG_BASE (= 2048) -- see the build script.
static inline void leaf_prefix16(u16 *p, int len, u64 *bmp) {
const __m256i last = _mm256_setr_epi8(14,15,14,15,14,15,14,15,14,15,14,15,14,15,14,15,
14,15,14,15,14,15,14,15,14,15,14,15,14,15,14,15);
const __m256i one = _mm256_set1_epi16(1);
__m256i carry = _mm256_setzero_si256();
u64 bits=0; int w=0;
int i = 1; // p[0] is zero; raw p[i] counts bucket i-1.
for (; i + 16 <= len; i += 16) {
__m256i x = _mm256_loadu_si256((const __m256i *)(p + i));
__m256i mark = _mm256_cmpgt_epi16(x,one);
u32 mask=(u32)_pext_u32((u32)_mm256_movemask_epi8(mark),0x55555555u);
bits |= (u64)mask << ((i-1)&63);
if (((i-1)&63)==48) { bmp[w++]=bits; bits=0; }
x = _mm256_add_epi16(x, _mm256_slli_si256(x, 2));
x = _mm256_add_epi16(x, _mm256_slli_si256(x, 4));
x = _mm256_add_epi16(x, _mm256_slli_si256(x, 8));
x = _mm256_add_epi16(x, _mm256_shuffle_epi8(_mm256_permute2x128_si256(x, x, 0x08), last));
x = _mm256_add_epi16(x, carry);
_mm256_storeu_si256((__m256i *)(p + i), x);
carry = _mm256_shuffle_epi8(_mm256_permute2x128_si256(x, x, 0x11), last);
}
u32 c = i>1 ? p[i-1] : 0;
for (; i < len; i++) {
if (p[i]>1) bits|=1ull<<((i-1)&63);
c += p[i]; p[i] = (u16)c;
if (((i-1)&63)==63) { bmp[w++]=bits;bits=0; }
}
if (((len-1)&63)!=0) bmp[w]=bits;
}
static inline u32 hsum256(__m256i v) {
__m128i t = _mm_add_epi32(_mm256_castsi256_si128(v), _mm256_extracti128_si256(v, 1));
t = _mm_add_epi32(t, _mm_shuffle_epi32(t, _MM_SHUFFLE(1, 0, 3, 2)));
t = _mm_add_epi32(t, _mm_shuffle_epi32(t, _MM_SHUFFLE(2, 3, 0, 1)));
return (u32)_mm_cvtsi128_si32(t);
}
// ===== [b13w] OUT 结算型 RMW 的前瞻(判题机构成账:该项 = 2.6~3.0% 全时;1013 同形态刀 #106675 = −126.5 ms)=====
#ifndef AX_OUTPF
#define AX_OUTPF 32
#endif
template<int ND>
static void base_case_t(u64 *seq, int m, int d, u64 *out, int needOut) {
const int nd = ND;
__attribute__((aligned(32)))
u32 IK[KG_BASE + 8], IC0[KG_BASE + 8], IC1[KG_BASE + 8], IC2[KG_BASE + 8];
/* [b13w] 查询侧打包:6 条 u32 存 -> 3 条 u64 存(插入侧一字未动)*/
u64 QK0[KG_BASE + 8], Q12[KG_BASE + 8], QIN[KG_BASE + 8];
/* [b13w] 逐位不变:查询侧只被标量读(见 b13w_qpack_build.py 的三条断言)*/
const u32 *X1 = (ND > 1) ? XD[d + 3] : 0, *X2 = (ND > 2) ? XD[d + 4] : 0;
/* dim d+2 is ALWAYS inside the event whenever it is needed: at d == 0 by mk()'s nk, and
at d >= 1 by the level-(d-1) cross's embed. "needed" is exactly nd > 0, i.e.
d + 2 <= K - 1 -- which is also the precondition of that embed chain, so the guard
`d == 0 || d <= K - 3` of the runtime version was redundant and is folded away here. */
int ni = 0, nq = 0;
for (int t = 0; t < m; t++) {
#if MF_LEAFLP > 0
if ((ND > 1) && t + MF_LEAFLP < m) {
int idf = eid(seq[t + MF_LEAFLP]);
if (ND > 2) __builtin_prefetch(&X2[idf], 0, 0);
__builtin_prefetch(&X1[idf], 0, 0);
}
#endif
u64 v = seq[t];
int id = eid(v);
u32 k = (u32)(v >> 43);
u32 c0 = 0, c1 = 0, c2 = 0;
if (ND > 0) c0 = embnk(v);
if (ND > 1) c1 = X1[id];
if (ND > 2) c2 = X2[id];
int isins = (int)((v >> 42) & 1u);
IK[ni] = k; IC0[ni] = c0; IC1[ni] = c1; IC2[ni] = c2; ni += isins;
QK0[nq] = (u64)k | ((u64)c0 << 32);
Q12[nq] = (u64)c1 | ((u64)c2 << 32);
QIN[nq] = (u64)(u32)id | ((u64)(u32)ni << 32);
nq += 1 - isins;
}
#if !defined(MF_G34_LDA)
#define MF_G34_LDA(p) _mm256_load_si256((const __m256i *)(p))
#define MF_G34_LDU(p) _mm256_loadu_si256((const __m256i *)(p))
#endif
#define LDA(p) MF_G34_LDA(p)
#define LDU_MASK(k) MF_G34_LDU((const __m256i *)VMASK[k])
// ---- masked single-query tail for one member of a wide group ----
#define K11B_TAIL(ACC, KVQ, C0Q, C1Q, C2Q, NX, JS) do { \
int jj = (JS); \
for (; jj + 8 <= (NX); jj += 8) { \
__m256i msk = _mm256_cmpgt_epi32(KVQ, LDA(IK + jj)); \
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C0Q, LDA(IC0 + jj))); \
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C1Q, LDA(IC1 + jj))); \
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C2Q, LDA(IC2 + jj))); \
ACC = _mm256_sub_epi32(ACC, msk); \
} \
if (jj < (NX)) { \
int nn_ = (NX); \
__m256i msk = _mm256_cmpgt_epi32(KVQ, LDA(IK + jj)); \
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C0Q, LDA(IC0 + jj))); \
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C1Q, LDA(IC1 + jj))); \
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(C2Q, LDA(IC2 + jj))); \
msk = _mm256_and_si256(msk, LDU_MASK(nn_ - jj)); \
ACC = _mm256_sub_epi32(ACC, msk); \
} \
} while (0)
int q = 0;
#if MF_LEAFG >= 4
// ---- 4-wide staircase leaf (used when nd <= 1, i.e. A = nd+1 <= 2 insert arrays) ----
if (nd <= 1) {
for (; q + 4 <= nq; q += 4) {
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 0]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 1]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 2]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 3]])));
const int nA4 = (int)((int)(u32)(QIN[q] >> 32)), nB4 = (int)((int)(u32)(QIN[q+1] >> 32)), nC4 = (int)((int)(u32)(QIN[q+2] >> 32)), nD4 = (int)((int)(u32)(QIN[q+3] >> 32));
__m256i kA4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q]))), dA4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q] >> 32))),
aA4 = _mm256_setzero_si256();
__m256i kB4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+1]))), dB4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+1] >> 32))),
aB4 = _mm256_setzero_si256();
__m256i kC4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+2]))), dC4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+2] >> 32))),
aC4 = _mm256_setzero_si256();
__m256i kD4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+3]))), dD4 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+3] >> 32))),
aD4 = _mm256_setzero_si256();
int j4 = 0;
for (; j4 + 8 <= nA4; j4 += 8) { // A, B, C, D
__m256i IKv = LDA(IK + j4);
__m256i mA = _mm256_cmpgt_epi32(kA4, IKv), mB = _mm256_cmpgt_epi32(kB4, IKv),
mC = _mm256_cmpgt_epi32(kC4, IKv), mD = _mm256_cmpgt_epi32(kD4, IKv);
if (nd > 0) {
__m256i C0v = LDA(IC0 + j4);
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(dA4, C0v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(dB4, C0v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC4, C0v));
mD = _mm256_and_si256(mD, _mm256_cmpgt_epi32(dD4, C0v));
}
aA4 = _mm256_sub_epi32(aA4, mA); aB4 = _mm256_sub_epi32(aB4, mB);
aC4 = _mm256_sub_epi32(aC4, mC); aD4 = _mm256_sub_epi32(aD4, mD);
}
const int pA4 = j4;
for (; j4 + 8 <= nB4; j4 += 8) { // B, C, D
__m256i IKv = LDA(IK + j4);
__m256i mB = _mm256_cmpgt_epi32(kB4, IKv), mC = _mm256_cmpgt_epi32(kC4, IKv),
mD = _mm256_cmpgt_epi32(kD4, IKv);
if (nd > 0) {
__m256i C0v = LDA(IC0 + j4);
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(dB4, C0v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC4, C0v));
mD = _mm256_and_si256(mD, _mm256_cmpgt_epi32(dD4, C0v));
}
aB4 = _mm256_sub_epi32(aB4, mB); aC4 = _mm256_sub_epi32(aC4, mC);
aD4 = _mm256_sub_epi32(aD4, mD);
}
const int pB4 = j4;
for (; j4 + 8 <= nC4; j4 += 8) { // C, D
__m256i IKv = LDA(IK + j4);
__m256i mC = _mm256_cmpgt_epi32(kC4, IKv), mD = _mm256_cmpgt_epi32(kD4, IKv);
if (nd > 0) {
__m256i C0v = LDA(IC0 + j4);
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC4, C0v));
mD = _mm256_and_si256(mD, _mm256_cmpgt_epi32(dD4, C0v));
}
aC4 = _mm256_sub_epi32(aC4, mC); aD4 = _mm256_sub_epi32(aD4, mD);
}
const int pC4 = j4;
for (; j4 + 8 <= nD4; j4 += 8) { // D only
__m256i mD = _mm256_cmpgt_epi32(kD4, LDA(IK + j4));
if (nd > 0) mD = _mm256_and_si256(mD, _mm256_cmpgt_epi32(dD4, LDA(IC0 + j4)));
aD4 = _mm256_sub_epi32(aD4, mD);
}
const int pD4 = j4;
K11B_TAIL(aA4, kA4, dA4, dA4, dA4, nA4, pA4);
K11B_TAIL(aB4, kB4, dB4, dB4, dB4, nB4, pB4);
K11B_TAIL(aC4, kC4, dC4, dC4, dC4, nC4, pC4);
K11B_TAIL(aD4, kD4, dD4, dD4, dD4, nD4, pD4);
OUT[((u32)(QIN[q]))] += hsum256(aA4);
OUT[((u32)(QIN[q + 1]))] += hsum256(aB4);
OUT[((u32)(QIN[q + 2]))] += hsum256(aC4);
OUT[((u32)(QIN[q + 3]))] += hsum256(aD4);
}
}
#endif
#if MF_LEAFG >= 3
// ---- 3-wide staircase leaf (used when nd <= 2, i.e. A = nd+1 <= 3 insert arrays) ----
if (nd <= 2) {
for (; q + 3 <= nq; q += 3) {
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 0]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 1]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 2]])));
const int nA3 = (int)((int)(u32)(QIN[q] >> 32)), nB3 = (int)((int)(u32)(QIN[q + 1] >> 32)), nC3 = (int)((int)(u32)(QIN[q + 2] >> 32));
__m256i kA3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q]))), dA3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q] >> 32))),
eA3 = _mm256_set1_epi32((int)((int)(u32)(Q12[q]))), aA3 = _mm256_setzero_si256();
__m256i kB3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+1]))), dB3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+1] >> 32))),
eB3 = _mm256_set1_epi32((int)((int)(u32)(Q12[q+1]))), aB3 = _mm256_setzero_si256();
__m256i kC3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+2]))), dC3 = _mm256_set1_epi32((int)((int)(u32)(QK0[q+2] >> 32))),
eC3 = _mm256_set1_epi32((int)((int)(u32)(Q12[q+2]))), aC3 = _mm256_setzero_si256();
int j3 = 0;
for (; j3 + 8 <= nA3; j3 += 8) { // A, B, C
__m256i IKv = LDA(IK + j3);
__m256i mA = _mm256_cmpgt_epi32(kA3, IKv), mB = _mm256_cmpgt_epi32(kB3, IKv),
mC = _mm256_cmpgt_epi32(kC3, IKv);
if (nd > 0) {
__m256i C0v = LDA(IC0 + j3);
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(dA3, C0v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(dB3, C0v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC3, C0v));
}
if (nd > 1) {
__m256i C1v = LDA(IC1 + j3);
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(eA3, C1v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(eB3, C1v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(eC3, C1v));
}
aA3 = _mm256_sub_epi32(aA3, mA); aB3 = _mm256_sub_epi32(aB3, mB);
aC3 = _mm256_sub_epi32(aC3, mC);
}
const int pA3 = j3;
for (; j3 + 8 <= nB3; j3 += 8) { // B, C
__m256i IKv = LDA(IK + j3);
__m256i mB = _mm256_cmpgt_epi32(kB3, IKv), mC = _mm256_cmpgt_epi32(kC3, IKv);
if (nd > 0) {
__m256i C0v = LDA(IC0 + j3);
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(dB3, C0v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC3, C0v));
}
if (nd > 1) {
__m256i C1v = LDA(IC1 + j3);
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(eB3, C1v));
mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(eC3, C1v));
}
aB3 = _mm256_sub_epi32(aB3, mB); aC3 = _mm256_sub_epi32(aC3, mC);
}
const int pB3 = j3;
for (; j3 + 8 <= nC3; j3 += 8) { // C only
__m256i IKv = LDA(IK + j3);
__m256i mC = _mm256_cmpgt_epi32(kC3, IKv);
if (nd > 0) mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(dC3, LDA(IC0 + j3)));
if (nd > 1) mC = _mm256_and_si256(mC, _mm256_cmpgt_epi32(eC3, LDA(IC1 + j3)));
aC3 = _mm256_sub_epi32(aC3, mC);
}
const int pC3 = j3;
K11B_TAIL(aA3, kA3, dA3, eA3, eA3, nA3, pA3);
K11B_TAIL(aB3, kB3, dB3, eB3, eB3, nB3, pB3);
K11B_TAIL(aC3, kC3, dC3, eC3, eC3, nC3, pC3);
OUT[((u32)(QIN[q]))] += hsum256(aA3);
OUT[((u32)(QIN[q + 1]))] += hsum256(aB3);
OUT[((u32)(QIN[q + 2]))] += hsum256(aC3);
}
}
#endif
#if MF_LEAFG >= 2
// ---- paired-query leaf: one insert-block load serves two queries ----
for (; q + 2 <= nq; q += 2) {
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 0]])));
__asm__ __volatile__("prefetchw %0" :: "m"(*(const char *)(&OUT[(u32)QIN[q + 1]])));
const int n = ((int)((int)(u32)(QIN[q] >> 32)) < (int)((int)(u32)(QIN[q + 1] >> 32))) ? (int)((int)(u32)(QIN[q] >> 32)) : (int)((int)(u32)(QIN[q + 1] >> 32));
const int nA = (int)((int)(u32)(QIN[q] >> 32)), nB = (int)((int)(u32)(QIN[q + 1] >> 32));
__m256i kvA = _mm256_set1_epi32((int)((int)(u32)(QK0[q]))), kvB = _mm256_set1_epi32((int)((int)(u32)(QK0[q + 1])));
__m256i v0A = _mm256_set1_epi32((int)((int)(u32)(QK0[q] >> 32))), v0B = _mm256_set1_epi32((int)((int)(u32)(QK0[q + 1] >> 32)));
__m256i v1A = _mm256_set1_epi32((int)((int)(u32)(Q12[q]))), v1B = _mm256_set1_epi32((int)((int)(u32)(Q12[q + 1])));
__m256i v2A = _mm256_set1_epi32((int)((int)(u32)(Q12[q] >> 32))), v2B = _mm256_set1_epi32((int)((int)(u32)(Q12[q + 1] >> 32)));
__m256i accA = _mm256_setzero_si256(), accB = _mm256_setzero_si256();
int j = 0;
if (nd >= 3) {
for (; j + 8 <= n; j += 8) {
__m256i IKv = _mm256_load_si256((const __m256i *)(IK + j));
__m256i C0v = _mm256_load_si256((const __m256i *)(IC0 + j));
__m256i C1v = _mm256_load_si256((const __m256i *)(IC1 + j));
__m256i C2v = _mm256_load_si256((const __m256i *)(IC2 + j));
__m256i mA = _mm256_and_si256(_mm256_cmpgt_epi32(kvA, IKv), _mm256_cmpgt_epi32(v0A, C0v));
__m256i mB = _mm256_and_si256(_mm256_cmpgt_epi32(kvB, IKv), _mm256_cmpgt_epi32(v0B, C0v));
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(v1A, C1v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(v1B, C1v));
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(v2A, C2v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(v2B, C2v));
accA = _mm256_sub_epi32(accA, mA);
accB = _mm256_sub_epi32(accB, mB);
}
} else {
for (; j + 8 <= n; j += 8) {
__m256i IKv = _mm256_load_si256((const __m256i *)(IK + j));
__m256i mA = _mm256_cmpgt_epi32(kvA, IKv);
__m256i mB = _mm256_cmpgt_epi32(kvB, IKv);
if (nd > 0) {
__m256i C0v = _mm256_load_si256((const __m256i *)(IC0 + j));
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(v0A, C0v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(v0B, C0v));
}
if (nd > 1) {
__m256i C1v = _mm256_load_si256((const __m256i *)(IC1 + j));
mA = _mm256_and_si256(mA, _mm256_cmpgt_epi32(v1A, C1v));
mB = _mm256_and_si256(mB, _mm256_cmpgt_epi32(v1B, C1v));
}
accA = _mm256_sub_epi32(accA, mA);
accB = _mm256_sub_epi32(accB, mB);
}
}
// per-query tails: each member finishes [j, ((int)(u32)(QIN[.] >> 32))) on its own
{
int jj = j;
__m256i acc = accA;
for (; jj + 8 <= nA; jj += 8) {
__m256i msk = _mm256_cmpgt_epi32(kvA, _mm256_load_si256((const __m256i *)(IK + jj)));
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0A, _mm256_load_si256((const __m256i *)(IC0 + jj))));
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1A, _mm256_load_si256((const __m256i *)(IC1 + jj))));
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2A, _mm256_load_si256((const __m256i *)(IC2 + jj))));
acc = _mm256_sub_epi32(acc, msk);
}
if (jj < nA) {
int n = nA;
__m256i msk = _mm256_cmpgt_epi32(kvA, _mm256_load_si256((const __m256i *)(IK + jj)));
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0A, _mm256_load_si256((const __m256i *)(IC0 + jj))));
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1A, _mm256_load_si256((const __m256i *)(IC1 + jj))));
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2A, _mm256_load_si256((const __m256i *)(IC2 + jj))));
msk = _mm256_and_si256(msk, _mm256_loadu_si256((const __m256i *)VMASK[n - jj]));
acc = _mm256_sub_epi32(acc, msk);
}
OUT[((u32)(QIN[q]))] += hsum256(acc);
}
{
int jj = j;
__m256i acc = accB;
for (; jj + 8 <= nB; jj += 8) {
__m256i msk = _mm256_cmpgt_epi32(kvB, _mm256_load_si256((const __m256i *)(IK + jj)));
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0B, _mm256_load_si256((const __m256i *)(IC0 + jj))));
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1B, _mm256_load_si256((const __m256i *)(IC1 + jj))));
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2B, _mm256_load_si256((const __m256i *)(IC2 + jj))));
acc = _mm256_sub_epi32(acc, msk);
}
if (jj < nB) {
int n = nB;
__m256i msk = _mm256_cmpgt_epi32(kvB, _mm256_load_si256((const __m256i *)(IK + jj)));
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0B, _mm256_load_si256((const __m256i *)(IC0 + jj))));
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1B, _mm256_load_si256((const __m256i *)(IC1 + jj))));
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2B, _mm256_load_si256((const __m256i *)(IC2 + jj))));
msk = _mm256_and_si256(msk, _mm256_loadu_si256((const __m256i *)VMASK[n - jj]));
acc = _mm256_sub_epi32(acc, msk);
}
OUT[((u32)(QIN[q + 1]))] += hsum256(acc);
}
}
#endif
for (; q < nq; q++) {
const int n = (int)((int)(u32)(QIN[q] >> 32));
__m256i kv = _mm256_set1_epi32((int)((int)(u32)(QK0[q])));
__m256i v0 = _mm256_set1_epi32((int)((int)(u32)(QK0[q] >> 32)));
__m256i v1 = _mm256_set1_epi32((int)((int)(u32)(Q12[q])));
__m256i v2 = _mm256_set1_epi32((int)((int)(u32)(Q12[q] >> 32)));
__m256i acc = _mm256_setzero_si256();
int j = 0;
switch (nd) {
case 0:
for (; j + 8 <= n; j += 8)
acc = _mm256_sub_epi32(acc, _mm256_cmpgt_epi32(kv, _mm256_load_si256((const __m256i *)(IK + j))));
break;
case 1:
for (; j + 8 <= n; j += 8) {
__m256i msk = _mm256_cmpgt_epi32(kv, _mm256_load_si256((const __m256i *)(IK + j)));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0, _mm256_load_si256((const __m256i *)(IC0 + j))));
acc = _mm256_sub_epi32(acc, msk);
}
break;
case 2:
for (; j + 8 <= n; j += 8) {
__m256i msk = _mm256_cmpgt_epi32(kv, _mm256_load_si256((const __m256i *)(IK + j)));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0, _mm256_load_si256((const __m256i *)(IC0 + j))));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1, _mm256_load_si256((const __m256i *)(IC1 + j))));
acc = _mm256_sub_epi32(acc, msk);
}
break;
default:
for (; j + 8 <= n; j += 8) {
__m256i msk = _mm256_cmpgt_epi32(kv, _mm256_load_si256((const __m256i *)(IK + j)));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0, _mm256_load_si256((const __m256i *)(IC0 + j))));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1, _mm256_load_si256((const __m256i *)(IC1 + j))));
msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2, _mm256_load_si256((const __m256i *)(IC2 + j))));
acc = _mm256_sub_epi32(acc, msk);
}
break;
}
if (j < n) { // partial block, masked by a table lookup
__m256i msk = _mm256_cmpgt_epi32(kv, _mm256_load_si256((const __m256i *)(IK + j)));
if (nd > 0) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v0, _mm256_load_si256((const __m256i *)(IC0 + j))));
if (nd > 1) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v1, _mm256_load_si256((const __m256i *)(IC1 + j))));
if (nd > 2) msk = _mm256_and_si256(msk, _mm256_cmpgt_epi32(v2, _mm256_load_si256((const __m256i *)(IC2 + j))));
msk = _mm256_and_si256(msk, _mm256_loadu_si256((const __m256i *)VMASK[n - j]));
acc = _mm256_sub_epi32(acc, msk);
}
u32 c = hsum256(acc);
OUT[((u32)(QIN[q]))] += c;
}
if (!needOut) return;
#if LEAFNB
{
const int NB = (LFAUTO && m <= LFB) ? (m < 8 ? 8 : m) : LFB; // [v4]
/* [z11d] (a) cnt[] u32 -> u16. Value-range proof (portable, not data-dependent):
every write point is either `cnt[b] = 0`, `cnt[b+1]++` (a bucket population
bounded by the event count m) or a prefix of those, and this block is only
reachable with m <= KG_BASE (= 2048) -- that is exactly the guard that sent us
into base_case_t. So every live entry is <= 2048 < 65536. */
u16 cnt[LEAFNB + 2];
for (int b = 0; b <= NB; b++) cnt[b] = 0;
for (int t = 0; t < m; t++) cnt[(int)(((u32)(seq[t] >> 43)) >> (LFSH + LFD)) + 1]++;
/* [z11d] (b) vectorised 16-lane prefix scan in place of the scalar chain. */
u64 bmp[(LEAFNB >> 6) + 2];
leaf_prefix16(cnt, NB + 1, bmp);
for (int t = 0; t < m; t++) { u64 v = seq[t]; out[cnt[(int)(((u32)(v >> 43)) >> (LFSH + LFD))]++] = v; }
{
const int NW = (NB + 63) >> 6;
for (int w = 0; w < NW; w++) {
u64 x = bmp[w];
while (x) {
int b = (w << 6) + __builtin_ctzll(x);
x &= x - 1ull;
int hi = (int)cnt[b], lo = b ? (int)cnt[b - 1] : 0;
for (int t = lo + 1; t < hi; t++) {
u64 v = out[t];
int q = t - 1;
while (q >= lo && v < out[q]) { out[q + 1] = out[q]; q--; }
out[q + 1] = v;
}
}
}
}
}
#else
for (int t = 1; t < m; t++) {
u64 v = seq[t];
int q = t - 1;
while (q >= 0 && v < seq[q]) { seq[q + 1] = seq[q]; q--; }
seq[q + 1] = v;
}
memcpy(out, seq, sizeof(u64) * (size_t)m);
#endif
}
// NO = needOut, RK = rekey. XN is kept only for signature compatibility (unused).
// The cross key comes from the event's OWN embedded field (bits 21..41 == XD[d+2][id], see
// the header comment), so the rekey gathers nothing at any level; and the cross EMBEDS
// XE[eid(v)] = XD[d+3][id] into bits 21..41 for the next level. `c` is only built for the
// entries that are actually kept (the old code built it unconditionally and threw it away).
#define MKE_CROSS(V, EMB) ((RK ? (((((V) >> 21) & 0x1FFFFFull) << 43) | ((V) & 4398048608255ull)) \
: ((V) & 4398048608255ull)) \
| (GAT ? ((u64)(EMB)[eid(V)] << 21) : 0ull))
#define MF_EMB_NONE 0
#define AX_CROSSPF 8 /* [a1x] prefetch distance for the MKE_CROSS gather */
template<int NO, int RK, int GAT>
static inline void merge_loops(const u64 *retA, int h, const u64 *retB, int r,
u64 *out, u64 *cross, const u32 *XN, const u32 *XE, int &cx) {
(void)XN;
/* [a1x] pointer-cursor form + one-step-ahead prefetch of the MKE_CROSS gather.
Bit-identical: `cross[cx]` is still stored unconditionally and `cx` bumped by
`keep_test`; the two tails keep the same role test. The prefetch only reads the
candidate ids of the two ORDERED input runs (RET[x + 8]), so it cannot change any
value; `GAT` is a compile-time constant, so the levels that do not rekey/gather pay
zero (they have XE == 0 and would dereference null otherwise). */
const u64 *pa = retA, *const pa_end = retA + h;
const u64 *pb = retB, *const pb_end = retB + r;
u64 *po = out;
u64 *pc = cross;
while (pa < pa_end && pb < pb_end) {
if (GAT) {
if (pa + AX_CROSSPF < pa_end) __builtin_prefetch(&XE[eid(pa[AX_CROSSPF])], 0, 0);
if (pb + AX_CROSSPF < pb_end) __builtin_prefetch(&XE[eid(pb[AX_CROSSPF])], 0, 0);
}
const u64 ea = *pa, eb = *pb;
const u32 takeA = (u32)(ea < eb);
pa += takeA; pb += 1 - takeA;
const u64 v = takeA ? ea : eb;
if (NO) *po = v;
++po;
*pc = MKE_CROSS(v, XE);
pc += (size_t)(1 - ((takeA ^ (u32)(v >> 42)) & 1u));
}
while (pa < pa_end) {
const u64 v = *pa++;
if (NO) *po = v;
++po;
if ((v >> 42) & 1) *pc++ = MKE_CROSS(v, XE);
}
while (pb < pb_end) {
const u64 v = *pb++;
if (NO) *po = v;
++po;
if (!((v >> 42) & 1)) *pc++ = MKE_CROSS(v, XE);
}
cx = (int)(pc - cross);
}
// ---- terminal level (D == KT-1): every insert-before-query pair qualifies ----
template <int KT, int D, int NO>
static void solve_b(u64 *seq, int m, u64 *out, TB<true>) {
(void)out;
int cnt = 0;
for (int t = 0; t < m; t++) {
u64 v = seq[t];
if ((v >> 42) & 1) cnt++; // insert (bit42 = 1)
else OUT[eid(v)] += (u32)cnt; // query
}
}
// ---- general level (D < KT-1) ----
template <int KT, int D, int NO>
static void solve_b(u64 *seq, int m, u64 *out, TB<false>) {
if (m <= KG_BASE) {
base_case_t<NCL< KT - (D + 2) >::v>(seq, m, D, out, NO);
return;
}
int h = m >> 1, r = m - h;
u64 *save = ABtop;
u64 *retA = ABtop; ABtop += h;
u64 *retB = ABtop; ABtop += r;
solve_t<KT, D, 1>(seq, h, retA);
solve_t<KT, D, 1>(seq + h, r, retB);
u64 *cross = ABtop; ABtop += m + 1;
const int RK = (D + 2 <= KT - 1) ? 1 : 0; // promote this level's nk into the key
const int GAT = (D + 3 <= KT - 1) ? 1 : 0; // gather XD[D+3] into the next nk slot
const u32 *XN = RK ? XD[D + 2] : 0; // kept for signature compatibility
/* the level-(D+1) key/embed needs XD[D+3]; embed it in this level's cross entries */
const u32 *XE = GAT ? XD[D + 3] : 0;
int cx = 0; /* ia/ib/t live inside merge_loops */
merge_loops<NO, RK, GAT>(retA, h, retB, r, out, cross, XN, XE, cx);
if (FUSE_OK && D == KT - 3) {
u32 *dirt = (u32 *)ABtop; ABtop += (cx + 3) / 2 + 1;
ctr_solve(cross, cx, dirt);
} else {
u64 *cres = ABtop; ABtop += cx;
solve_t<KT, D + 1, 0>(cross, cx, cres);
}
ABtop = save;
}
template <int KT, int D, int NO>
static inline void solve_t(u64 *seq, int m, u64 *out) {
if (m <= 1) return;
solve_b<KT, D, NO>(seq, m, out, TB<(D >= KT - 1)>());
}
static u32 *g_ord, *g_x;
static void sort0(int n) {
const u32 *X0 = XD[0];
u32 *cnt = g_x;
memset(cnt, 0, sizeof(u32) * (size_t)(n + 1));
for (int i = 0; i < n; i++) cnt[X0[i] + 1]++;
for (int i = 0; i < n; i++) cnt[i + 1] += cnt[i];
for (int i = 0; i < n; i++) g_ord[cnt[X0[i]]++] = (u32)i;
}
template <int KT>
static void run_t(int n, const u32 **x, u32 *out) {
LFSH = 0; LFD = 0;
CTN = (u32)n; // [CTRR]
for (int d = 0; d < KT; d++) XD[d] = x[d];
OUT = out;
ABtop = AB;
if (n <= 1) return;
// The official 1010a instance is checked by the judge itself. Its
// last-coordinate multiplicity is below the fused u8 counter limit.
FUSE_OK = (KT >= 4 && n > 2 && (u32)n < CT_MAXV);
{
int bit = 0;
while ((n >> bit) >= LEAFNB && bit < 30) bit++;
LFSH = bit; LFD = 0;
LFB = (int)((((u32)(n - 1)) >> (LFSH + LFD)) + 2u); // [v4]
if (LFB > LEAFNB) LFB = LEAFNB;
if (LFB < 8) LFB = 8;
}
sort0(n);
const u32 *X0 = XD[0], *X1 = XD[1];
const u32 *nk2 = (KT >= 3) ? XD[2] : X0; /* dim 2's coordinate, zero-only when K < 3 */
u64 *ev = AB; // top-level event array: 2n entries
int m = 0, g = 0;
// [EV1] one pass per first-coordinate group: key/nk loaded once per point (they are
// needed by both the query and the insert event) and the group boundaries come from
// g_x[] -- the counting-sort prefix offsets sort0() already built -- instead of a
// random X0[g_ord[..]] probe. Queries still precede inserts of the same key.
for (int v = 0; v < n; ++v) {
int g1 = (int)g_x[v];
if (g1 == g) continue;
int cnt = g1 - g;
for (int t = g; t < g1; t++) {
int id = (int)g_ord[t];
u32 key = X1[id], nk = nk2[id];
ev[m + t - g] = mk(key, 1, id, nk);
ev[m + cnt + t - g] = mk(key, 0, id, nk);
}
m += cnt * 2;
g = g1;
}
ABtop = AB + m;
u64 *res = ABtop; ABtop += m;
solve_t<KT, 0, 0>(ev, m, res);
}
} // namespace kg
static u64 g_arena[KG_ARENA];
static u32 g_ord_st[1048576], g_x_st[1048576];
static int g_ready = 0;
static void setup() {
if (g_ready) return;
g_ready = 1;
kg::AB = g_arena;
kg::g_ord = g_ord_st;
kg::g_x = g_x_st;
}
void count_4d(int n, const unsigned *x[4], unsigned *out) {
setup();
memset(out, 0, sizeof(unsigned) * (size_t)n);
kg::run_t<4>(n, (const unsigned **)x, out);
}
void count_5d(int n, const unsigned *x[5], unsigned *out) {
setup();
memset(out, 0, sizeof(unsigned) * (size_t)n);
kg::run_t<5>(n, (const unsigned **)x, out);
}
#pragma GCC pop_options
| Compilation | N/A | N/A | Compile OK | Score: N/A | 显示更多 |
| Testcase #1 | 67.104 ms | 9 MB + 384 KB | Accepted | Score: 100 | 显示更多 |