BZOJ1562 [NOI2009]变换序列【KM算法】

时间：2018-04-26 21:10:57 阅读：189 评论：0 收藏：0 [点我收藏+]

标签：oid 数据 code math als memset 输出 char 一个

题目

技术分享图片

输入格式

技术分享图片

输出格式

技术分享图片

输入样例

1 1 2 2 1

输出样例

1 2 4 0 3

提示

30%的数据中N≤50；
60%的数据中N≤500；
100%的数据中N≤10000。

题解

每个位置可以和两种数匹配，显然是一个二分图匹配问题

但要求字典序最小，我们就按字典序存边
由于在KM算法中，后来着具有更高的优先级，我们倒序匹配即可

#include<iostream>
#include<cstdio>
#include<cmath>
#include<vector>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<‘ ‘; puts("");
using namespace std;
const int maxn = 10005,maxm = 100005,INF = 1000000000;
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57){if (c == ‘-‘) flag = -1; c = getchar();}
    while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    return out * flag;
}
int lk[maxn],vis[maxn],ans[maxn];
int h[maxn],ne = 1;
struct EDGE{int to,nxt;}ed[maxn << 1];
inline void build(int u,int v){
    ed[ne] = (EDGE){v,h[u]}; h[u] = ne++;
}
bool find(int u){
    Redge(u) if (!vis[to = ed[k].to]){
        vis[to] = true;
        if (lk[to] == -1 || find(lk[to])){
            lk[to] = u;
            return true;
        }
    }
    return false;
}
int n,a[maxn],b[maxn];
int main(){
    memset(lk,-1,sizeof(lk));
    n = read(); int tmp,a,b;
    for (int i = 0; i < n; i++){
        tmp = read();
        a = (i + tmp) % n;
        b = (i - tmp + n) % n;
        if (a < b) swap(a,b);
        build(i,a); build(i,b);
    }
    for (int i = n - 1; i >= 0; i--){
        memset(vis,0,sizeof(vis));
        if (!find(i)){
            puts("No Answer");
            return 0;
        }
    }
    for (int i = 0; i < n; i++) ans[lk[i]] = i;
    for (int i = 0; i < n; i++){
        printf("%d",ans[i]);
        if (i < n - 1) printf(" ");
    }
    return 0;
}