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层
    }
} 

上一篇:有理数均值


下一篇:10-python的re模块的使用(1)