当前位置: 代码迷 >> 综合 >> 【NOI2009】bzoj1562 变换序列
  详细解决方案

【NOI2009】bzoj1562 变换序列

热度:55   发布时间:2024-01-13 10:45:29.0

可以把距离看成每个点都有两个匹配点,这样就是求字典序最小的二分图匹配。先任意求出一个,然后从小到大考虑每个点,尝试更换他的匹配点,但是要求不能改动之前的匹配。

#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]);
}