ZOJ 3954 Seven-Segment Display

二分图匹配。

先检查每个数字$1$的个数是否满足条件，不满足直接就是无解。剩下的情况可以建立二分图，如果现在的某一列可以对应于原图的某一列，那么建边。如果二分图的最大匹配是$7$，则有解，否则误解。

#include<bits/stdc++.h>
using namespace std;

int n;
int num[],g[][];
char s[][];
int c[];
int cx[], cy[];
int mk[],nx,ny;

int path(int u)
{
    for (int v = ; v<ny; v++)
    {
        if (g[u][v] && !mk[v])
        {
            mk[v] = ;
            if (cy[v] == - || path(cy[v]))
            {
                cx[u] = v;
                cy[v] = u;
                return ;
            }
        }
    }
    return ;
}

int MaxMatch()
{
    int res = ;
    memset(cx, -, sizeof(cx));
    memset(cy, -, sizeof(cy));
    for (int i = ; i<nx; i++)
    {
        if (cx[i] == -)
        {
            memset(mk, , sizeof(mk));
            res = res + path(i);
        }
    }
    return res;
}

char m[][]={
    "",
    "",
    "",
    "",
    "",
    "",
    "",
    "",
    "",
    ""
};

void init()
{
    c[]=;
    c[]=;
    c[]=;
    c[]=;
    c[]=;
    c[]=;
    c[]=;
    c[]=;
    c[]=;
}

int main()
{
    init();  int T;
    scanf("%d",&T);
    int fail;
    while(T--)
    {
        fail=;
        scanf("%d",&n);
        for(int i=;i<=n;i++) scanf("%d%s",&num[i],s[i]);
        for(int i=;i<=n;i++)
        {
            int sum=;
            for(int j=;s[i][j];j++)
            {
                if(s[i][j]=='') sum++;
            }
            if(sum!=c[num[i]]) fail=;
        }

        if(fail)
        {
            printf("NO\n");
            continue;
        }

        for(int i=;i<=;i++)
        {
            for(int j=;j<=;j++)
            {
                g[i][j]=;
            }
        }

        for(int i=;i<=n;i++)
        {
            for(int j=;s[i][j];j++)
            {
                for(int k=;k<m[num[i]][k];k++)
                {
                    if(s[i][j]!=m[num[i]][k]) g[j][k]=;
                }
            }
        }

        nx=,ny=;

        int ans = MaxMatch();

        if(ans!=) printf("NO\n");
        else printf("YES\n");

    }
    return ;
}