[素数筛+最大独立子集] Lightoj1356 Prime Independence

题目

题目

思路

题目要求一个数组内有多少对数互质或者除数不为质数。
可以反过来，找有多少对数的除数为质数即可。
先预处理每个数的质因子个数。
两个质因子个数为奇数和两个质因子个数为偶数的数都不会产生除数为质数的情况。所以可以建立一个二分图，一边的除数质因子个数为奇数，一边为偶数。如果它们除数为质数就相连，求他们的最大匹配。那么最后的答案就是n-最大匹配数

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
int a[40010],prime[500010];
bool vis[600010];
int tot1=0;
void get_prime(int n){
	for(int i=2;i<=n;i++){
		if(!vis[i]){
			prime[tot1++]=i;	
		}
		for(int j=0;j<tot1&&prime[j]*i<=n;j++){
			vis[prime[j]*i]=1;
			if(i%prime[j]==0) break;
		}
	}
	
}
struct sut{
	int to,next,value;
}edge[150010];
int tot=1;
int head[40010],dis[40010],cnt[40010];
int n,m,s,t;
void add_edge(int u,int v,int w){
	edge[++tot].to=v;
	edge[tot].value=w;
	edge[tot].next=head[u];
	head[u]=tot;
}
bool bfs(){
	for(int i=0;i<=t;i++){
		dis[i]=-1;
	}
	queue<int> q;
	dis[s]=0;
	q.push(s);
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=head[u];i;i=edge[i].next){
			int to=edge[i].to;
			if(dis[to]==-1&&edge[i].value>0){
				dis[to]=dis[u]+1;
				q.push(to);
			}
		}
	}
	return dis[t]!=-1;
}
int dfs(int x,int exp){
	if(x==t) return exp;
	int flow=0;
	for(int i=head[x];i;i=edge[i].next){
		int to=edge[i].to;
		if((dis[to]==dis[x]+1)&&edge[i].value>0){
			int tmp=dfs(to,min(exp,edge[i].value));
			if(!tmp) continue;
			exp-=tmp;
			flow+=tmp;
			edge[i].value-=tmp;
			edge[i^1].value+=tmp;
			if(!exp) break;
		}
	}
	if(!flow) dis[x]=-2;
	return flow;
}
vector<int> fac[40010]; 
int main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int tt;
	cin>>tt;
	get_prime(500000);
	//for(int i=0;i<tot;i++) cout<<prime[i]<<" \n"[i==tot-1];
	int num=0;
	while(tt--){
		num++;
		int n;
		cin>>n;
		tot=1;
		for(int i=0;i<=n+1;i++) head[i]=0,cnt[i]=0,fac[i].clear();
		s=0;
		t=n+1;
		
		map<int,int> mp;
		for(int i=1;i<=n;i++){
			cin>>a[i];
			int tmp=a[i];
			mp[a[i]]=i; 
			for(int j=0;prime[j]*prime[j]<=tmp;j++){
				if(tmp%prime[j]==0){
					fac[i].push_back(prime[j]);
					tmp/=prime[j];
					cnt[i]++;
					while(tmp%prime[j]==0)	tmp/=prime[j],cnt[i]++;
				}
			}
			if(tmp>1) fac[i].push_back(tmp),cnt[i]++; 
			if(cnt[i]&1){
				add_edge(s,i,1);
				add_edge(i,s,0);
			}
			else{
				add_edge(i,t,1);
				add_edge(t,i,0);
			}
		}
		for(int i=1;i<=n;i++){
			for(int j=0;j<fac[i].size();j++){
				int tmp=a[i]/fac[i][j];
				if(mp[tmp]){
					if(cnt[i]&1){
						add_edge(i,mp[tmp],1);
						add_edge(mp[tmp],i,0);
					}
					else{
						add_edge(i,mp[tmp],0);
						add_edge(mp[tmp],i,1);
					}
				}
			}
		}
		int ans=0;
		while(bfs()){
			ans+=dfs(s,INF);
		}
		cout<<"Case "<<num<<": "<<n-ans<<endl;
	}
    return 0;
}