题目描述
本题译自 eJOI2018 Problem F「Cycle Sort」
给定一个长为 \(n\) 的数列 \(\{a_i\}\) ,你可以多次进行如下操作:
选定 \(k\) 个不同的下标 \(i_1, i_2, \cdots, i_k\)(其中 \(1 \le i_j \le n\) ),然后将 \(a_{i_1}\) 移动到下标 \(i_2\) 处,将 \(a_{i_2}\) 移动到下标 \(i_3\) 处,……,将 \(a_{i_{k-1}}\) 移动到下标 \(i_{k}\) 处,将 \(a_{i_k}\) 移动到下标 \(i_1\) 处。
换言之,你可以按照如下的顺序轮换元素:\(i_1 \rightarrow i_2 \rightarrow i_3 \rightarrow \cdots \rightarrow i_{k-1} \rightarrow i_k \rightarrow i_1\) 。
例如:\(n=4, \{a_i\}=\{ 10, 20, 30, 40\}, i_1=2, i_2=3, i_3=4\) ,则操作完成后的 \(a\) 数列变为 \(\{ 10, 40, 20, 30\}\) 。
你的任务是用操作次数最少的方法将整个数列排序成不降的。注意,所有操作中选定下标的个数总和不得超过 \(s\) 。如果不存在这样的方法(无解),输出 -1。
输入格式
第一行, \(2\) 个整数, \(n\) 和 \(s\ (1 \le n \le 2 \times 10^5, 0 \le s \le 2 \times 10^5)\) 。
第二行, \(n\) 个整数 \(a_1, a_2, a_3, \cdots a_n\) ,表示数列 \(\{a_i\}\) ,其中 \(1 \le a_i \le 10^9\) 。
输出格式
如果无解,仅输出 -1 。
否则,第一行输出一个整数 \(q\) (可以为 \(0\) ,参见样例 3 ),表示最少需要进行的操作次数。
接下来 \(2 \times q\) 行描述每次操作。
对于每次操作,先输出一个整数 \(k\) 表示此操作选定的下标数量,然后在下一行中输出 \(k\) 个整数 \(i_1, i_2, \cdots, i_k\) 。
在操作次数 \(q\) 最少的情况下,你可以输出任意一种可行方案。
样例
样例输入 1
5 5
3 2 3 1 1
样例输出 1
1
5
1 4 2 3 5
样例解释 1
你可以用两次操作 \(1 \rightarrow 4 \rightarrow 1\) 和 \(2 \rightarrow 3 \rightarrow 5 \rightarrow 2\) 排序数组,但这样会 WA,因为你的任务是最小化 \(q\) ,而最优解的 \(q=1\) 。
一种可行的方法是 \(1 \rightarrow 4 \rightarrow 2 \rightarrow 3 \rightarrow 5 \rightarrow 1\) ,即样例输出。
样例输入 2
4 3
2 1 4 3
样例输出 2
-1
样例解释 2
所有操作中选定下标的个数总和的最小值为 \(4\) (一种可行的方法是 \(1 \rightarrow 2 \rightarrow 1\) 和 \(3 \rightarrow 4 \rightarrow 3)\),因此无解。
样例输入 3
2 0
2 2
样例输出 3
0
样例解释 3
数组已经有序,因此不需要进行操作。
样例输入 4
6 9
6 5 4 3 2 1
样例输出 4
2
6
1 6 2 5 3 4
3
3 2 1
样例输入 5
6 8
6 5 4 3 2 1
样例输出 5
3
2
3 4
4
1 6 2 5
2
2 1
补充说明
样例 4 和 5 满足子任务 6 和 7 的限制。
数据范围与提示
子任务编号 | 分数 | 限制 |
---|---|---|
1 | $0$ | 样例 |
$2$ | $5$ | $n, s \le 2$ 且 $1 \le a_i \le 2$ |
$3$ | $5$ | $n \le 5$ |
$4$ | $5$ | $1 \le a_i \le 2$ |
$5$ | $1$$0$ | $\{a_i\}$ 中 $1$ 到 $n$ 的所有正整数出现且恰好只出现一次, $s=2 \times n$ |
$6$ | $10$ | $\{a_i\}$ 中 $1$ 到 $n$ 的所有正整数出现且恰好只出现一次, $n \le 1000$ |
$7$ | $15$ | $\{a_i\}$ 中 $1$ 到 $n$ 的所有正整数出现且恰好只出现一次 |
$8$ | $15$ | $s=2 \times n$ |
$9$ | $15$ | $n \le 1000$ |
$10$ | $20$ | 无特殊限制 |
题解
首先假设没有相同的 \(a_i\)
离散化一下,对于每个点,如果它错位,那么都把它向它本应该在的地方连边。我们就可以得到很多个环,如果没有要求要操作数最小,那么一种方案就是每个环移动一回,直至最后完成
如果错位个数大于允许操作次数,那么肯定是无解的了
特判掉只有一个环的情况
然后有一种和并环的方法,就是把上面得到的所有的环首尾相接连在一起,最后会发现只有每个环的最后一个位置错位,我们再把这些错位的连成一个新的环操作,所以最少次数就是 \(2\) 次
但是这样的移动次数是 错位个数 \(+\) 环数 的,很有可能会超过允许的操作次数,这个时候,我们可以选择不把所有的环合并成一个大环,我们可以把某一些环合并成大环,按照大环的做法去做,而剩下的小环就一个一个自己换,这样可以减少移动次数
所以最后就这样贪心选就好了
如果有相同的 \(a_i\) 也是一样的,同样的离散化,但找环是在一个欧拉图上找的,直接找欧拉回路
#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXN=200000+10;
int n,s,k,e,a[MAXN],b[MAXN],beg[MAXN],nex[MAXN],to[MAXN],val[MAXN],ext[MAXN],vis[MAXN],use[MAXN],cnt;
std::vector<int> V,ans[MAXN];
std::map<int,int> M;
template<typename T> inline void read(T &x)
{
T data=0,w=1;
char ch=0;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
if(x<0)putchar('-'),x=-x;
if(x>9)write(x/10);
putchar(x%10+'0');
if(ch!='\0')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void discretization()
{
for(register int i=1;i<=n;++i)V.push_back(a[i]);
std::sort(V.begin(),V.end());
V.erase(std::unique(V.begin(),V.end()),V.end());
for(register int i=0,lt=V.size();i<lt;++i)M[V[i]]=i+1;
}
inline void insert(int x,int y,int z)
{
to[++e]=y;
nex[e]=beg[x];
beg[x]=e;
val[e]=z;
}
inline void dfs(int x)
{
vis[x]=cnt;
for(register int &i=beg[x];i;i=nex[i])
if(!use[i])
{
int tmp=i;
use[i]=1;
dfs(to[i]);
ans[cnt].push_back(val[tmp]);
}
}
int main()
{
read(n);read(s);
for(register int i=1;i<=n;++i)read(a[i]);
discretization();
for(register int i=1;i<=n;++i)a[i]=b[i]=M[a[i]];
std::sort(b+1,b+n+1);
for(register int i=1;i<=n;++i)
if(a[i]!=b[i])++k,insert(a[i],b[i],i);
if(k>s)
{
puts("-1");
return 0;
}
for(register int i=1;i<=n;++i)
if(!vis[i]&&beg[i])++cnt,dfs(i);
if(cnt<=1||s-k<=1)
{
printf("%d\n",cnt);
for(register int i=1;i<=cnt;++i)
{
printf("%d\n",ans[i].size());
for(register int j=0,lt=ans[i].size();j<lt;++j)printf("%d ",ans[i][j]);
puts("");
}
return 0;
}
printf("%d\n",cnt-min(s-k,cnt)+2);
for(register int i=s-k+1;i<=cnt;++i)
{
printf("%d\n",ans[i].size());
for(register int j=0,lt=ans[i].size();j<lt;++j)printf("%d ",ans[i][j]);
puts("");
}
if(s-k)
{
int p1=0,p2=0;
for(register int i=min(s-k,cnt);i>=1;--i)p1+=ans[i].size(),ext[++p2]=ans[i][0];
printf("%d\n",p1);
for(register int i=1;i<=min(s-k,cnt);++i)
for(register int j=0,lt=ans[i].size();j<lt;++j)printf("%d ",ans[i][j]);
puts("");
printf("%d\n",p2);
for(register int i=1;i<=p2;++i)printf("%d ",ext[i]);
}
return 0;
}