BZOJ 3140 HNOI2013 消毒

Problem

BZOJ

Solution

不妨先考虑二维平面上的消除方法。则思想类似于HEOI2016的游戏，对于一个需要消除的点(x,y)，我们对x,y进行连线，然后最后跑最大匹配就是答案。

那么考虑三维情况，由于x*y*y<=5000，则必有x,y,z其中一个数值小于$\sqrt[3] {5000}≈17$，那么不妨对于最小的一个方向上进行dfs枚举。

Code

#include <cstring>
#include <cstdio>
using namespace std;
template <typename Tp> inline void read(Tp &x)
{
    x=0;int f=0;char ch=getchar();
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') f=1,ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    if(f) x=-x;
}
const int INF=0x3f3f3f3f;
struct data{int v,w,nxt;}edge[5010];
int z,n,p,pos,dfn,lim,ans,d[5],id[5010][5];
int head[5010],link[5010],vis[5010],t[25];
inline int min(int x,int y){return x<y?x:y;}
inline int max(int x,int y){return x>y?x:y;}
inline void insert(int u,int v,int w){edge[++p]=(data){v,w,head[u]};head[u]=p;}
void input()
{
    int x;
    read(d[1]);read(d[2]);read(d[3]);
    ans=INF;lim=n=pos=p=0;memset(head,0,sizeof(head));
    for(int i=1;i<=3;i++) if(d[pos]>d[i]) pos=i;
    for(int i=1;i<=3;i++) if(i!=pos) lim=max(d[i],lim);
    for(int i=1;i<=d[1];i++)
      for(int j=1;j<=d[2];j++)
        for(int k=1;k<=d[3];k++)
        {
            read(x);
            if(x) id[++n][1]=i,id[n][2]=j,id[n][3]=k;
        }
    for(int i=1;i<=n;i++)
    {
        if(pos==1) insert(id[i][2],id[i][3],id[i][1]);
        else if(pos==2) insert(id[i][1],id[i][3],id[i][2]);
        else insert(id[i][1],id[i][2],id[i][3]);
    }
}
inline int find(int u)
{
    for(int i=head[u];i;i=edge[i].nxt)
      if(!t[edge[i].w]&&vis[edge[i].v]!=dfn)
      {
        vis[edge[i].v]=dfn;
        if(!link[edge[i].v]||find(link[edge[i].v]))
        {
            link[edge[i].v]=u;
            return 1;
        }
      }
    return 0;
}
int check(int x)
{
    int res=0;
    memset(link,0,sizeof(link));
    for(int i=1;i<=lim;i++)
    {
        dfn++;
        if(find(i)) res++;
        if(x+res>=ans) break;
    }
    return x+res;
}
void dfs(int x,int dep)
{
    if(dep>d[pos]){ans=min(ans,check(x));return ;}
    t[dep]=1;dfs(x+1,dep+1);
    t[dep]=0;dfs(x,dep+1);
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    scanf("%d",&z);
    d[0]=INF;
    while(z--)
    {
        input();
        dfs(0,1);
        printf("%d\n",ans);
    }
    return 0;
}