洛谷1963 [noi2009]变换序列

匈牙利算法的本质

阅读全文大概需要 4分钟
本文总阅读量
Posted by yjjr's blog on October 29, 2018

标签:匈牙利算法

题目

题目传送门

题目描述

对于\(N\)个整数\(0, 1, \cdots, N-1\),一个变换序列\(T\)可以将\(i\)变成\(T_i\),其中 \(T_i \in \{ 0,1,\cdots, N-1\}\) 且 \(\bigcup_{i=0}^{N-1} \{T_i\} = \{0,1,\cdots , N-1\}\)。 ,\(\forall x,y \in \{0,1,\cdots , N-1\}\),定义x和y之间的距离\(D(x,y)=min\{\|x-y\|,N-\|x-y\|\}\) 。给定每个\(i\)和\(T_i\)之间的距离\(D(i,T_i)\),你需要求出一个满足要求的变换序列T。如果有多个满足条件的序列,输出其中字典序最小的一个。

说明:对于两个变换序列\(S\)和\(T\),如果存在\(p<N\),满足对于\(i=0,1,\cdots p-1\),\(S_i=T_i\)且\(S_p<T_p\),我们称\(S\)比\(T\)字典序小。

输入输出格式

输入格式

第一行包含一个整数\(N\),表示序列的长度。接下来的一行包含\(N\)个整数\(D_i\),其中\(D_i\)表示\(i\)和\(T_i\)之间的距离。

输出格式

如果至少存在一个满足要求的变换序列\(T\),则输出文件中包含一行\(N\)个整数,表示你计算得到的字典序最小的\(T\);否则输出No Answer(不含引号)。注意:输出文件中相邻两个数之间用一个空格分开,行末不包含多余空格。

输入输出样例

输入样例#1

5
1 1 2 2 1

输出样例#1

1 2 4 0 3

说明

对于30%的数据,满足:N<=50;

对于60%的数据,满足:N<=500;

对于100%的数据,满足:N<=10000。

分析

二分图的完美匹配问题

每个点只与\((d_i+i)\%n\),\((n-d_i+i)\%n\)联通

因为匈牙利算法的本质,是当不匹配的情况下,后面的点把前面的点挤出的过程,所以倒着扫描匹配可以解决字典序最小的问题

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read(){
	ll f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
//******head by yjjr******
const int maxn=1e4+6;
int n,d[maxn],mt[maxn],ans[maxn],last[maxn<<2],cnt=0;bool vis[maxn];
inline int dis(int x,int y){return min(abs(x-y),n-abs(x-y));}
vector <int> e[maxn];
inline bool dfs(int x){
	for(int i=0;i<e[x].size();i++)
		if(!vis[e[x][i]]){
			vis[e[x][i]]=1;
			if(mt[e[x][i]]==0||dfs(mt[e[x][i]])){mt[e[x][i]]=x;return 1;}
		}
	return 0;
}
int main(){
	n=read();
	rep(i,0,n-1)d[i]=read();
	rep(i,0,n-1){
		int x=i+d[i],y=i-d[i]+n;
		x%=n,y%=n;
		if(dis(x,i)!=d[i])x=-1;
		if(dis(y,i)!=d[i])y=-1;
		if(x>y)swap(x,y);
		if(x!=-1)e[i].push_back(x);
		if(y!=-1)e[i].push_back(y);
	}
	dep(i,n-1,0){
		mem(vis,0);
		if(!dfs(i)){puts("No Answer");return 0;}
	}
	rep(i,0,n-1)ans[mt[i]]=i;
	rep(i,0,n-1)cout<<ans[i]<<' ';cout<<endl;
	return 0;
}
本文可以转载,但必须附上原文链接,否则你会终生找不到妹子!!!欢迎关注我的CSDN: ahyjjr