图论杂题

Sort with Swap(0,*) (25)

Link
这道题要求一定是和0交换。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <string>
#include <string.h>
#include <vector>
#include <cmath>
using namespace std;
int n,cnt;
int a[100000];
bool vis[100000];
int main() {
	scanf("%d",&n);
	for(int i=0;i<n;++i)
		scanf("%d",&a[i]);
	for(int i=0;i<n;++i)
		if(!vis[i]){
			vis[i]=true;
			int cur=a[i],tmp=1;
			while(cur!=i){
				vis[cur]=true;
				cur=a[cur];
				tmp++;
			}
			if(tmp==1) continue;
			if(i==0) cnt+=tmp-1;
			else cnt+=tmp+1;
		}
	printf("%d\n",cnt);
	return 0;
}

如果不是和0交换，而是任意两数交换都可以，那么只用改以下部分即可：

for(int i=0;i<n;++i)
	if(!vis[i]){
		vis[i]=true;
		int cur=a[i],tmp=1;
		while(cur!=i){
			vis[cur]=true;
			cur=a[cur];
			tmp++;
		}
		cnt+=tmp-1;
	}