题解:Atcoder Beginner Contest abc466 F - Many Mod Calculation

题目描述

给你整数 N,X 和长度为 N 的正整数序列 A=(A \_ 1,A \_ 2,\\ldots,A \_ N)

对于一个非负整数 x ,定义 f(x)=(\\ldots((x \\bmod A \_ 1) \\bmod A \_ 2) \\ldots ) \\bmod A \_ N

求介于 1X 之间的整数 x 的个数,使得 f(x)=0 .

给你 T 个测试用例,请逐个求解。

数据范围

  • 1\\le T\\le 2\\times 10\^5
  • 1\\le N\\le 2\\times 10\^5
  • 所有测试用例中 N 的总和最多为 2\\times 10\^5
  • 1\\le X\\le 10\^{18}
  • 1\\le A_i\\le 10\^{18}
  • 所有输入值均为整数。

解题思路

核心思想

这个问题要求统计在 [1, X] 范围内,经过一系列取模操作后结果为 0 的数的个数。

关键观察:取模操作具有"分段常数"性质 。对于给定的模数 v,所有数 x 在区间 [kv, (k+1)v - 1] 内取模 v 的结果相同,都是 x - kv

算法设计

用一个**优先队列(最大堆)**来维护区间信息,每个元素表示一个"状态区间":

  • val:当前状态值(取模后的结果)

  • cnt:有多少个原始数 x 能达到这个状态

具体步骤

  1. 初始化

    复制代码
    pq.push({x + 1, 1});  // 表示原始区间 [0, X] 内有 X+1 个数,状态值为 X+1
  2. 逐个处理模数

    对每个 v = A[i],将当前所有状态 valv 取模:

    • 从堆中取出最大的 val(因为我们需要处理所有相同的 val

    • 合并所有相同 val 的计数

    • 分裂成新状态:

      • {v, val / v * cnt}:表示能整除的部分,取模后结果为 0

      • {val % v, cnt}:表示余数部分(如果余数不为 0)

  3. 统计答案

    最终堆中所有 val = 0 的状态计数之和,就是 f(x) = 0x 的数量。

    由于统计的是 [0, X] 区间(包含 0),而题目要求 [1, X],所以答案为 总计数 - 1(减去 x = 0 的情况)。

完整代码

复制代码
#include<bits/stdc++.h>
#define fr1(i,a,b) for(int (i)=(a);(i)<=(b);(i)++)
#define fr2(i,a,b) for(int (i)=(a);(i)>=(b);(i)--)
#define fv(i,p) for(auto (i):(p))
#define ll long long
#define ull unsigned ll
#define pii pair<int,int>
#define pll pair<ll,ll>
#define _1st first
#define _2nd second
#define elif else if
#define debug cout<<endl<<"-------------------------------------------------------------"<<endl
using namespace std;

void solve() {
    int n;
    ll x;
    cin >> n >> x;
    vector<ll> a(n);
    for (ll &v : a) cin >> v;
    
    priority_queue<pll> pq;
    pq.push({x + 1, 1});
    
    for (ll v : a) {
        while (!pq.empty() && pq.top()._1st > v) {
            auto [val, cnt] = pq.top();
            pq.pop();
            while (!pq.empty() && pq.top()._1st == val) {
                cnt += pq.top()._2nd;
                pq.pop();
            }
            pq.push({v, val / v * cnt});
            if (val % v != 0) pq.push({val % v, cnt});
        }
    }
    
    ll ans = -1;
    while (!pq.empty()) {
        ans += pq.top()._2nd;
        pq.pop();
    }
    cout << ans << '\n';
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int t;
    cin >> t;
    while (t--) solve();
    
    return 0;
}
相关推荐
Chester_19995 小时前
CSP202203C.计算资源调度器
开发语言·数据结构·c++·蓝桥杯
会周易的程序员6 小时前
给 PLC 写一个字节码虚拟机:STVM 虚拟机架构设计
开发语言·c++·虚拟机·软plc·iec61131·stvm
小灰灰搞电子7 小时前
完全驾驭 Qt 与数据库:C++ ORM 框架 QxOrm 原理与实践指南
数据库·c++·qt
QT界面美化性能优化7 小时前
QT+AI:使用AI技术为QT应用程序赋能
c++·人工智能·qt·opencv·qt教程·qt6.3
学习星球9 小时前
单调栈——从“找下一个更大的“到柱状图中的最大矩形
数据库·c++·算法·leetcode·xcode
小灰灰搞电子10 小时前
C++ 引用折叠详解
c++
布莱克60510 小时前
数组详解:定义、作用、应用场景及与链表的区别(C/C++ 代码讲解)
c语言·开发语言·c++·数组
牛油果子哥q11 小时前
C++内存池与对象池精讲:内存碎片、自定义分配器、对象池实现、STL allocator原理、工程落地与性能对比
java·开发语言·c++
万法若空12 小时前
C/C++ 类型别名定义方式对比
java·c语言·c++
小小晓.13 小时前
C++ 值类别与完美转发深度剖析:从左值右值到 std::forward 源码解密
开发语言·c++·算法