AcWing每日一题 3636. 数组延伸

原题链接

给定一个初始长度为 n 的数组 a 以及一个整数 x。

我们现在要对数组 a 进行延伸，具体方法如下：

我们从数组中的第一个元素开始，逐个遍历数组中的每个元素。

当遍历到数组中的元素 q 时，如果 q 能够被 x 整除，则在数组的末尾添加 x 个整数 qx，并开始遍历下一个元素。

否则，停止遍历，数组延伸结束。

注意，后面新增的元素也要被考虑在内，加以处理和判断。

请计算，在数组延伸结束后，数组中所有元素的和。

数据范围 1≤T≤100, 1≤n≤105, 2≤x≤109, 1≤ai≤109, 输入保证，所有 T 个 n 的和不超过 105。

输入格式
第一行包含整数 T，表示共有 T 组测试数据。

每组数据的第一行包含两个整数 n 和 x。

第二行包含 n 个整数 a1,a2,…,an。

输出格式
每组数据输出一行，一个结果，表示延伸结束后，数组中所有元素的和。

输入样例：
2
1 2
12
4 2
4 6 8 2
输出样例：
36
44
样例解释
第一组数据，最终数组为 [12,6,6,3,3,3,3]。

第二组数据，最终数组为 [4,6,8,2,2,2,3,3,4,4,1,1,1,1,1,1]。

首先注意题目是遍历到不能整除x的数为止，就不继续遍历了，且每次拆分一个数，拆完的数加起来还是等于这个数。
所以我们只需要处理出能够遍历的最小层数（即第一次出现不能整除x的数的层数），前面的值就是最初数组的和再乘以层数（不包含最后一层），再加上最后一层的数即可

AC代码：

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int maxn=1e5+7;       
int n;
int x;
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d %d",&n,&x);
        int k=maxn;
        ll suma,sumb;//suma是初始数组和
        suma=sumb=0;
        for(int i=1;i<=n;++i){
            int a;
            scanf("%d",&a);
            int tt=0;
            suma+=a;
            for(int j=a;j&&j%x==0;j/=x)tt++;//计算该数所能到的层数
            if(tt<k){
                sumb=suma-a;
                k=tt;
            }
        }
        printf("%lld\n",suma*(k+1)+sumb);//k+1是加的是0层
    }
} 

标签：遍历,3636,元素,int,数组,sumb,suma,AcWing
来源： https://blog.csdn.net/qq_42671102/article/details/118436065