GYM 101243 J. Architect of Your Own Fortune【二分匹配】

给出N个A数以及M个T数，让我们凑出最多的对数，使得放在前边的数的前三位的加和以及放在后边的数的后三位的数的加和是相等的。

如果T数放在前边，就是TA数，否则就是AT数 。

思路：

暴力二分匹配+输出路径。

Ac代码：

#include <bits/stdc++.h>
typedef long long int LL;
using namespace std;
void fre(){
    freopen("INPUT.TXT","r",stdin);
    freopen("OUTPUT.TXT","w",stdout);
}
int n,m;
int match[1500];
int vis[1500];
char a[150][7];
char b[150][7];
vector<int>mp[1500];
int Calpre(int i)
{
    int sum=0;
    for(int j=0;j<3;j++)
    {
        sum+=a[i][j]-'0';
    }
    return sum;
}
int Calback(int i)
{
    int sum=0;
    for(int j=3;j<6;j++)
    {
        sum+=a[i][j]-'0';
    }
    return sum;
}
int Calpre2(int i)
{
    int sum=0;
    for(int j=0;j<3;j++)
    {
        sum+=b[i][j]-'0';
    }
    return sum;
}
int Calback2(int i)
{
    int sum=0;
    for(int j=3;j<6;j++)
    {
        sum+=b[i][j]-'0';
    }
    return sum;
}
int find(int u)
{
    for(int i=0;i<mp[u].size();i++)
    {
        int v=mp[u][i];
        if(vis[v]==0)
        {
            vis[v]=1;
            if(match[v]==-1||find(match[v]))
            {
                match[v]=u;
                return 1;
            }
        }
    }
    return 0;
}
int main()
{
    fre();
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n+m;i++)mp[i].clear();
        for(int i=1;i<=n;i++)scanf("%s",a[i]);
        for(int i=1;i<=m;i++)scanf("%s",b[i]);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                if(Calpre(i)==Calback2(j))
                {
                    mp[i].push_back(j+n);
                    mp[j+n].push_back(i);
                }
                else if(Calback(i)==Calpre2(j))
                {
                    mp[i].push_back(j+n);
                    mp[j+n].push_back(i);
                }
            }
        }
        int output=0;
        memset(match,-1,sizeof(match));
        map<pair<int,int>,int>s;
        for(int i=1;i<=n+m;i++)
        {
            memset(vis,0,sizeof(vis));
            if(find(i))output++;
        }
        printf("%d\n",output/2);
        for(int i=1;i<=n;i++)
        {
            if(match[i]==-1)continue;
            else
            {
                if(s[make_pair(i,match[i])]==0)
                {
                    int u=i;
                    int v=match[i]-n;
                    if(Calpre(u)==Calback2(v))
                    {
                        printf("AT %s %s\n",a[u],b[v]);
                    }
                    else
                    {
                        printf("TA %s %s\n",b[v],a[u]);
                    }
                    s[make_pair(i,match[i])]=1;
                    s[make_pair(match[i],i)]=1;
                }
            }
        }
    }
}