题解: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;
}
相关推荐
ShineWinsu40 分钟前
对于C++中unordered_map的详细介绍
数据结构·c++·算法·面试·stl·哈希表·unordered_map
东东最爱敲键盘44 分钟前
day3.c++函数与预处理 string
c++·算法
lingran__1 小时前
C++ 高阶数据结构:AVL 树万字详解|原理、四种旋转、key‑value 实现、调试校验、set/map/红黑树 底层前置知识
数据结构·c++·二叉搜索树·平衡二叉树·avl树·红黑树前置
爱和冰阔落2 小时前
【Linux】两个毫无关系的进程怎么通信?命名管道 FIFO 从原理到 Server/Client 实战
android·linux·数据库·c++·vim
码匠许师傅9 小时前
【C++ 面试真题】C++ 的 const 和 constexpr 有什么区别?
c++·面试
果果燕12 小时前
实习笔记(一):NFS、Samba、VNC、Git、CMake、systemd、内核、交叉编译
linux·arm开发·c++·笔记·git
牛艺翔13 小时前
C++基础
开发语言·c++
键盘会跳舞13 小时前
C++ :容器适配器stack源码级拆解
开发语言·c++··stack·先进后出
不负岁月无痕13 小时前
简单理解操作系统结构
java·linux·c语言·开发语言·c++·面试
余额瞒着我当琳14 小时前
C++模板初阶与STL初探
android·java·c++