scoi小凸玩矩阵 匈牙利+二分

来源:互联网 发布:php ob get contents 编辑:程序博客网 时间:2024/05/21 08:55

http://www.lydsy.com/JudgeOnline/problem.php?id=4443


参考leijp思想:

每行只选一个,每一列只选一个,我们把每行当成左边,把每一列当成右边,每一行和一列只会有一条边。

然后,把矩阵的值放在边上,比如(i,j)的值放在左边i行 到 右边j列上。

这里要求第K大(没注意,以为是第K小。  然后K=n-K+1,求第K小)


我们二分枚举最后的答案,然后使所有选择的边权值都不超过mid,看最多能匹配多少。  如果匹配数>=k,说明这个值的范围可以更小,    反之,这个值需要更大。


然后建图,二分+匈牙利就可以了。

(初学读入优化QAQ)

#include<cstdio>#include<iostream>#include<cstring>#include<queue>#include<cstdlib>#include<algorithm>#include<cmath>#include<bitset>using namespace std;const int maxn=255;struct edge{int v,w,next;}e[maxn*maxn];int head[maxn];int match[maxn*2];//bitset<maxn*2>vis;bool vis[maxn*2];int cnt;int n,m,k;void add(int u,int v,int w){e[cnt].v=v;e[cnt].w=w;e[cnt].next=head[u];head[u]=cnt++;}int read(){int res=0;int f=1;char ch=getchar();for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;for(;ch>='0'&&ch<='9';ch=getchar())res=res*10+ch-'0';return res*f;return res;}bool dfs(int u,int w){for(int i=head[u];~i;i=e[i].next){if(e[i].w>w)continue;int v=e[i].v;if(!vis[v]){vis[v]=1;if(match[v]==-1||dfs(match[v],w)){match[v]=u;return true;}}}return false;}bool check(int w){memset(match,-1,sizeof(match));int ans=0;for(int i=1;i<=n;i++){//vis=0;memset(vis,0,sizeof(vis));if(dfs(i,w))ans++;}return ans>=k;}int main(){memset(head,-1,sizeof(head));cnt=0;n=read(),m=read(),k=read();k=n-k+1;int l=0x3f3f3f3f,r=-0x3f3f3f3f;for(int i=1;i<=n;i++){for(int j=1;j<=m;j++){int x=read();add(i,j+n,x);l=min(l,x);r=max(r,x);}}while(l<r){int mid=(l+r)>>1;if(check(mid))r=mid;//>=kelse l=mid+1;}printf("%d\n",l);return 0;}


0 0
原创粉丝点击