题解: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;
}
相关推荐
吹什么轩2 小时前
c++复习:map和set的使用
开发语言·c++
fpcc2 小时前
跟我学C++中级篇—内存流
开发语言·c++
Tisfy3 小时前
LeetCode 3345.最小可整除数位乘积 I:暴力枚举(从n开始尝试)
数学·算法·leetcode·题解·枚举
吹什么轩3 小时前
c++复习:模拟实现list来更好的理解和应用list
开发语言·c++
AmyLin_20014 小时前
PDF 色彩保真工程实践【5】核心实现(下):拼装 Image XObject、页面资源与内容流
c++·pdf·sdk·颜色空间·印刷·icc·cmyk
乱七八糟的屋子4 小时前
MTL4 (Matrix Template Library) 超详细实战教程|C++高性能稠密/稀疏矩阵计算
c++·矩阵·稀疏矩阵·mtl4·matrixtemplate·c++线性代数·高性能数值计算
依然鸣6 小时前
PTA团体程序设计天梯赛L2真题讲解L2-045-048
数据结构·c++·经验分享·学习·算法·pat考试·pat
有点。6 小时前
C++深度优先搜索(二)
开发语言·c++·深度优先
思麟呀7 小时前
C++(七):std::any+std::string_view
开发语言·c++
小小龙学IT7 小时前
Taskflow:用一张“任务图“玩转现代 C++ 并行编程开源项
c++·开源·github