牛客练习赛53 德育分博弈政治课 【网络流】

题目大意

给出n个只有六个面的骰子
每个面由1~9 中的数字组成
每个面的数字不同
给出q次询问
每次给出一个串，
要求用n个骰子摆出串的形状

题目分析

因为每个骰子只能表示一个面
类似一个由限制的问题
考虑网络流中的一个点

如果它的进入的流量为1 就能限制它在与它相连的点中只选择一个。

那么我们考虑建这样一个图

每个数字到汇点的权值应该是查询串中这个数字出现的次数

然后，只要我们能跑一个满流，问题就可以解决了

但是给出的n太大了。有5e5 ，是没有办法跑网络流的。

但是我们发现所有的骰子的种类数只有C(9,6)
所以，我们可以发现，对骰子进行压缩，把相同的骰子看作一个点，边权是这种骰子的个数，那么这个图就可以缩小了

代码详解

#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 5e5+50,M = 5e6+50;
int n,m,s,t,tot,maxflow;
int head[M],ver[M],edge[M],nxt[M],d[N]; 
queue<int>q;
void add(int x,int y,int z)
{
    ver[++tot] = y,edge[tot]=z,nxt[tot]=head[x],head[x]=tot;
    ver[++tot] = x,edge[tot]=0,nxt[tot]=head[y],head[y]=tot;
}
bool bfs()
{
    memset(d,0,sizeof(d));
    while(q.size()) q.pop();
    q.push(s); 
    d[s]=1;
    while(!q.empty())
    {
        int x= q.front();q.pop();
        for(int i=head[x];i;i=nxt[i])
        {
            if(edge[i]&&!d[ver[i]])
            {
                q.push(ver[i]);
                d[ver[i]] = d[x]+1;
                if(ver[i]==t) return 1; 
            } 
        }
    }
    return 0;
}
int dinic(int x,int flow)
{
    if(x==t) return flow;
    int rest = flow,k;
    for(int i=head[x];i&&rest;i=nxt[i])
    {
        if(edge[i] &&d[ver[i]]==d[x]+1)
        {
            k = dinic(ver[i],min(rest,edge[i]));
            if(!k) d[ver[i]] = 0;
            edge[i]-=k;
            edge[i^1] +=k;
            rest-=k;    
        } 
    } 
    return flow-rest;
}
int cnt[10];
char ch[N][7],query[M];
int idx = 0;
map<int,int>mp;
int num[5000];
int id=0;
void getcnt(char x[M])
{
    memset(cnt,0,sizeof(cnt));
    int len = strlen(x);
    for(int i=0;i<len;i++)
    {
        cnt[x[i]-'0'] ++;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        char s[7];
        scanf("%s",s);
        int w = 0;
        for(int j=0;j<6;j++)
        {
            int t = s[j] - '0';
            w|=(1<<t);
        }
        if(mp[w]!=0)
        {
            num[mp[w]]++;
        }
        else
        {
            mp[w] = ++id;
            num[id] = 1;
            strcpy(ch[id],s);
        }
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%s",query);
        getcnt(query);
        int len = strlen(query);
        if(len>n) 
        {
            puts("zzk");
            continue;
        }
        tot = 1; s=0;t=n+20;
        memset(head,0,sizeof(head));
        memset(nxt,0,sizeof(nxt));
        int idx = 10;
        for(int j=1;j<=9;j++)
        {
            if(cnt[j]);
            add(j,t,cnt[j]);
        }
        for(int j=1;j<=id;j++)
        {
            //cout<<2<<endl;
            add(s,idx,num[j]);
            for(int k=0;k<6;k++)
            {
                int p = ch[j][k]-'0';
                add(idx,p,num[j]);
            }
            idx++;
        }
        maxflow = 0;
        int flow = 0;
        while(bfs())
        {
            while(flow = dinic(s,inf)) maxflow +=flow;
        }
        if(maxflow == len) puts("dyf");
        else puts("zzk");
    }
    return 0;
}

注：
代码已修改