原题链接:https://codeforc.es/problemset/problem/1348/B
题意:告诉我们一个数组及其长度和k,判断是否可以构造一个新数组使得每K段长度和都相等。
思路:首先需要判断原数组不同元素的个数与k的大小,如果看k小于原数组的不同元素的个数那么无法构造这样的数组;如果k大于等于那我们就可以构造出一个新数组。我们定义一个vis[101]数组作为标记数组,记录不同元素的个数,来判断是否存在这样的数组。然后我们定义一个数组b[]构造新数组的一个周期数组。最后只要输出n个这样的周期数组即可。
坑点:一开始觉得很难 是以为新数组的长度要最短,然后反复看了几遍才发现(英语不好的菜鸡加上读题不仔细)。然后别忘记对标记数组的初始化,因为这个也wr了一次。
Ac代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod1=1e9+7;
const ll mod2=0x3f3f3f3f;
const ll maxn=1e4+5;
int a[maxn];//原数组;
int b[maxn];//新数组;
int vis[101];//标记数组;
int main(){
int t,n,k;
cin>>t;
while(t--){
cin>>n>>k;
for(int i=0;i<101;i++) vis[i]=0;//标记数组初始化;
for(int i=0;i<n;i++){cin>>a[i];vis[a[i]]++;}
int cnt=0;
for(int i=1;i<101;++i) {if(vis[i]) cnt++;}//求不同元素的个数;
if(cnt>k) {puts("-1");continue;}
else if(cnt<k){
for(int i=1;i<=100;i++){
if(!vis[i]){
vis[i]=1;cnt++;
if(cnt==k) break;//以k为周期;
}
}
}
int x=0;
for(int i=1;i<=100;i++){
if(vis[i]) b[++x]=i;//构造以k为周期的新数组;
}
cout<<n*k<<endl;//输出新数组的长度;
for(int i=0;i<n;i++){
for(int j=1;j<=k;j++){
cout<<b[j]<<" "; //输出新数组的元素;
}
}
cout<<endl;
}
return 0;
}