B. Phoenix and Beauty
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
Phoenix loves beautiful arrays. An array is beautiful if all its subarrays of length k have the same sum. A subarray of an array is any sequence of consecutive elements.
Phoenix currently has an array a of length n. He wants to insert some number of integers, possibly zero, into his array such that it becomes beautiful. The inserted integers must be between 1 and n inclusive. Integers may be inserted anywhere (even before the first or after the last element), and he is not trying to minimize the number of inserted integers.
Input
The input consists of multiple test cases. The first line contains an integer t (1≤t≤50) — the number of test cases.
The first line of each test case contains two integers n and k (1≤k≤n≤100).
The second line of each test case contains n space-separated integers (1≤ai≤n) — the array that Phoenix currently has. This array may or may not be already beautiful.
Output
For each test case, if it is impossible to create a beautiful array, print -1. Otherwise, print two lines.
The first line should contain the length of the beautiful array m (n≤m≤104). You don’t need to minimize m.
The second line should contain m space-separated integers (1≤bi≤n) — a beautiful array that Phoenix can obtain after inserting some, possibly zero, integers into his array a. You may print integers that weren’t originally in array a.
If there are multiple solutions, print any. It’s guaranteed that if we can make array a beautiful, we can always make it with resulting length no more than 104.
Example
inputCopy
4
4 2
1 2 2 1
4 3
1 2 2 1
3 2
1 2 3
4 4
4 3 4 2
outputCopy
5
1 2 1 2 1
4
1 2 2 1
-1
7
4 3 2 1 4 3 2
Note
In the first test case, we can make array a beautiful by inserting the integer 1 at index 3 (in between the two existing 2s). Now, all subarrays of length k=2 have the same sum 3. There exists many other possible solutions, for example:
2,1,2,1,2,1
1,2,1,2,1,2
In the second test case, the array is already beautiful: all subarrays of length k=3 have the same sum 5.
In the third test case, it can be shown that we cannot insert numbers to make array a beautiful.
In the fourth test case, the array b shown is beautiful and all subarrays of length k=4 have the same sum 10. There exist other solutions also.
思路
假如
6 5
1 2 2 3 4，那麼1 2 3 4 1 1 2 3 4 1 1 2 3 4 1，即可，也就是說以k爲單位，1 2 3 4 1 連續出現n次就能一定滿足情況，題目並未要求所求出的m爲最小值。
但是
6 3
1 2 2 3 4不可能滿足。
#include <iostream>
#include<cstdio>
#include<cstring>
#include<set>
#include<string>
#include<algorithm>
using namespace std;
set<int>p;
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
cin>>t;
while(t--)
{
int n,k;
cin>>n>>k;
int a;
if(n==k) cout<<n<<endl;
for(int i=1;i<=n;i++)
{
cin>>a;
p.insert(a);
if(n==k)cout<<a<<" ";
}
if(n==k)
{
p.clear();
cout<<endl;
continue;
}
else if((int)p.size()>k)
{
cout<<"-1"<<endl;
p.clear();
continue;
}
else
{
int m=k/(int)p.size();
int mm=k-m*((int)p.size());
set<int>::iterator it;
vector<int>ans;
while(m--)
{
for(it=p.begin();it!=p.end();it++)
{
ans.push_back(*it);
}
}
it=p.begin();
for(int j=0;j<mm;j++)
{
ans.push_back(*it);
it++;
}
cout<<n*k<<endl;
for(int i=1;i<=n;i++)
{
for(int i=0;i<k;i++)
cout<<ans[i]<<" ";
}
cout<<endl;
p.clear();
}
}
return 0;
}