可以把距离看成每个点都有两个匹配点,这样就是求字典序最小的二分图匹配。先任意求出一个,然后从小到大考虑每个点,尝试更换他的匹配点,但是要求不能改动之前的匹配。
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=10010;
int a[maxn],fir[maxn*2],ne[maxn*4],to[maxn*4],f[maxn],vis[maxn],ans[maxn],n,num;
void add(int u,int v)
{
ne[++num]=fir[u];
fir[u]=num;
to[num]=v;
}
int dfs1(int u)
{
int v;
if (vis[u]) return 0;
vis[u]=1;
for (int i=fir[u];i;i=ne[i])
if (f[v=to[i]]==-1||dfs1(f[v]))
{
f[v]=u;
return 1;
}
return 0;
}
int dfs2(int u,int from)
{
int v;
if (vis[u]) return 0;
vis[u]=1;
for (int i=fir[u];i;i=ne[i])
if (f[v=to[i]]==from||(f[v]>from&&dfs2(f[v],from)))
{
f[v]=u;
return 1;
}
return 0;
}
int main()
{
int x,y;
scanf("%d",&n);
for (int i=0;i<n;i++)
{
scanf("%d",&a[i]);
x=(i+a[i])%n;
y=(i-a[i]+n)%n;
if (x<y) swap(x,y);
add(i,x);
if (x!=y) add(i,y);
}
for (int i=0;i<n;i++) f[i]=-1;
for (int i=0;i<n;i++)
{
for (int j=0;j<n;j++) vis[j]=0;
if (!dfs1(i))
{
printf("No Answer");
return 0;
}
}
for (int i=0;i<n;i++)
{
for (int j=0;j<n;j++) vis[j]=0;
dfs2(i,i);
}
for (int i=0;i<n;i++) ans[f[i]]=i;
printf("%d",ans[0]);
for (int i=1;i<n;i++) printf(" %d",ans[i]);
}