原题链接: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; }