P2065 [TJOI2011]卡片(弧优化dinic)

P2065 [TJOI2011]卡片 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

桌子上现在有m张蓝色卡片和n张红色卡片，每张卡片上有一个大于1的整数。现在你要从桌子上拿走一些卡片，分若干次拿。每次只能拿走一组卡片：这组卡片颜色不同，并且两张卡片上面的数字的最大公约数大于1。问：最多可以从桌上拿走多少张卡片。

输入格式

每个输入文件中包含多组测试数据，每个文件中测试数据的数目不超过100。

文件的第一行读入一个整数T，为数据组数。

每组数据的格式如下：

m n

b1 b2 … bm

r1 r2 … rn

第二行给出每张蓝色卡片上面的数字，第三行给出每张红色卡片上的数字。

输出格式

对每组测试数据，输出最多可以拿走多少张卡片。

#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
int red,blue,t;
const int N=202;
const int M=5005;
int head[500000];
int cur[500000];
struct node{
    int to;
    int nxt;
    ll w;
}edge[500000];
int tot;
int dis[350000];
int source=0,sink;
int MAP[350000];
int d[350000];
int cnt;
void add(int from,int to,ll w)
{
    edge[tot].to=to;
    edge[tot].w=w;
    edge[tot].nxt=head[from];
    head[from]=tot++;
    edge[tot].to=from;
    edge[tot].w=0;
    edge[tot].nxt=head[to];
    head[to]=tot++;
}
void divide_blue(int x,int id)
{
    int y=0;
    for(int i=2;i<=sqrt(x);i++)
    {
        if(x%i==0)
        {
            d[++y]=i;
            while(x%i==0)x/=i;
        }
    }
    if(x>1)d[++y]=x;
    for(int i=1;i<=y;i++)
    {
        if(MAP[d[i]]==0){
            MAP[d[i]]=++cnt;
            add(id,red+blue+MAP[d[i]],1ll);
        }
        else add(id,red+blue+MAP[d[i]],1ll);
    }
}
void divide_red(int x,int id)
{
    int y=0;
    for(int i=2;i<=sqrt(x);i++)
    {
        if(x%i==0)
        {
            d[++y]=i;
            while(x%i==0)x/=i;
        }
    }
    if(x>1)d[++y]=x;
    for(int i=1;i<=y;i++)
    {
        if(MAP[d[i]]!=0){
            add(red+blue+MAP[d[i]],id+blue,1ll);
        }
    }
}
bool bfs()
{
    memset(dis,-1,sizeof(dis));
    dis[source]=0;
    queue<int> que;
    que.push(source);
    cur[source]=head[source];
    while(!que.empty())
    {
        int v=que.front();
        que.pop();
        if(v==sink)return true;
        for(int i=head[v];~i;i=edge[i].nxt)
        {
            int now=edge[i].to;
            ll cost=edge[i].w;
            if(dis[now]<0&&cost>0)
            {
            	cur[now]=head[now];
                dis[now]=dis[v]+1;
                que.push(now);
            }
        }

    }
    return false;
}
ll dfs(int v,ll flow)
{
    if(v==sink)return flow;
    ll pre=0;
    for(int i=cur[v];~i&&pre<flow;i=edge[i].nxt)
    {
    	cur[v]=i;
        int now=edge[i].to;
        ll cost=edge[i].w;
        if(dis[v]+1==dis[now]&&cost)
        {
            ll temp=dfs(now,min(cost,flow-pre));
            edge[i].w-=temp;
            edge[i^1].w+=temp;
            pre+=temp;
            if(temp==0)dis[now]=-1;
        }
    }
    if(pre==0)dis[v]=-1;
    return pre;
}
ll dinic()
{
    ll ans=0;
    ll flow=0;
    while(bfs())ans+=dfs(source,0x3f3f3f3f3f3f3f3f);
    return ans;
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>t;
    int x;
    while(t--)
    {
        cnt=tot=0;
        memset(head,-1,sizeof(head));
        memset(MAP,0,sizeof(MAP));
        cin>>blue>>red;
        for(int i=1;i<=blue;i++)
        {
            add(0,i,1ll);
        }
        for(int i=1;i<=blue;i++){
            cin>>x;
            divide_blue(x,i);
        }
        for(int i=1;i<=red;i++)
        {
            cin>>x;
            divide_red(x,i);
        }
        for(int i=1;i<=red;i++)add(i+blue,1+red+blue+cnt,1ll);
        sink=1+red+blue+cnt;
        cout<<dinic()<<endl;
    }
    return 0;
}