打卡信奥刷题（1509）用C++实现信奥 P5956 [POI 2017] Podzielno-优快云博客

P5956 [POI 2017] Podzielno

题目描述

$B$ 进制数，每个数字 $\in [0,B)$ 有 $a_i$ 个。你要用这些数字组成一个最大的 $B$ 进制数 $X$ (不能有前导零，不需要用完所有数字)，使得 $X$ 是 $B - 1$ 的倍数。 $q$ 次询问，每次询问 $X$ 在 $B$ 进制下的第 $k$ 位数字是什么(最低位是第 $0$ 位)。

输入格式

第一行包含两个正整数 $B, q$ 。

第二行包含 $B$ 个正整数 $a_0,a_1,a_2,...,a_{B-1}$ 。

接下来 $q$ 行，每行一个整数 $k$ ，表示一个询问。

输出格式

输出 $q$ 行，每行一个整数，依次回答每个询问，如果那一位不存在，请输出 -1。

输入输出样例 #1

输入 #1

输出 #1

0
2 
-1

说明/提示

对于 $100%100\%$ 的数据， $2≤B≤1062\le B\le10^6$ ， $1≤q≤1051\le q\le 10^5$ ， $1≤ai≤1061\le a_i\le10^6$ ， $0≤k≤10180\le k\le10^{18}$ 。

C++实现

#include <iostream>
#include <vector>
#include <algorithm>
int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int n, q;
    std::cin >> n >> q;
    std::vector<long long> a(n);
    long long sum = 0;
    for (int i = 0; i < n; ++i) {
        std::cin >> a[i];
        sum += i * a[i];
    }
    if (sum % (n - 1) != 0)
        --a[sum % (n - 1)];
    for (int i = 1; i < n; ++i)
        a[i] += a[i - 1];
    for (int i = 0; i < q; ++i) {
        long long k;
        std::cin >> k;
        if (k >= a[n - 1]) {
            std::cout << -1 << "\n";
        } else {
            std::cout << std::upper_bound(a.begin(), a.end(), k) - a.begin() << "\n";
        }
    }
    return 0;
}