给定一个初始长度为 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层
}
}