欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

距离序列(二分图匹配+字典序输出)

程序员文章站 2022-06-20 08:59:38
...

传送门

先暴力打出对应位置所能匹配的数,因为要最小字典序输出,所以匈牙利算法逆着来一遍,最后还可以在DFS中再加一个逆着匹配的数组,这就是答案了。


#include<bits/stdc++.h>

using namespace std;

const int maxn=1e4+50;

vector<int>edges[maxn];
bool vis[maxn];
int boy[maxn];
int gril[maxn];
int a[maxn];
int n;

bool dfs(int v)
{
    int len=edges[v].size();
    for(int i=0;i<len;i++){
        int end=edges[v][i];
        if(!vis[end]){
            vis[end]=1;
            if(boy[end]==0||dfs(boy[end])){
                boy[end]=v;
                gril[v]=end;
                return 1;
            }
        }
    }
    return 0;
}

int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(min(abs(i-j),n-abs(i-j))==a[i]){
                edges[i].push_back(j);
            }
        }
    }
    int ans=0;
    for(int i=n-1;i>=0;i--){
        memset(vis,0,sizeof(vis));
        if(dfs(i)){
            ans++;
        }
    }
    if(ans!=n){
        printf("No Answer\n");
    }else{
        for(int i=0;i<n-1;i++){
            printf("%d ",gril[i]);
        }
        printf("%d\n",gril[n-1]);
    }
    return 0;
}

 

 

相关标签: graph