Codeforces Round #638 (Div. 2) B. Phoenix and Beauty 题解（构造）

题目链接

题目思路

其实自己差不多想到这个方面，但是还是没有想透彻

为了使某个数组的k美观，该数组必须是周期为k的周期。 如果数组a中存在超过k个不同的数字，则没有答案，我们将打印-1（因为数组不能以周期k为周期）。 否则，我们提出以下构造。

考虑数组a中所有不同数字的列表。 如果少于k，我们将附加一些1（或任何其他数字），直到列表的大小为k。 我们可以将此列表打印n次。 我们的数组b的长度是nk，永远不超过1e4。可以始终通过在数组a中插入一些数字来构造数组b，因为a中的每个数字都对应一个列表。

每个测试用例的时间复杂度：O（nlogn + nk）

代码

#include <bits/stdc++.h>
using namespace std;

void solve(){
  int N,K;
  cin>>N>>K;
  set<int>s;
  for (int i=0;i<N;i++){
    int a;
    cin>>a;
    s.insert(a);
  }
  //if more than K distinct numbers, print -1
  if (s.size()>K){
    cout<<-1<<endl;
    return;
  }
  cout<<N*K<<endl;
  for (int i=0;i<N;i++){
    //print the distinct numbers
    for(set<int>::iterator ite=s.begin();ite!=s.end();ite++)
        cout<<*ite<<' ';
    //print the extra 1s
    for (int j=0;j<K-(int)s.size();j++)
        cout<<1<<' ';
  }
  cout<<endl;
}

int main(){
  int t; cin>>t;
  while (t--)
    solve();
}