4514: [Sdoi2016]数字配对
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 726 Solved: 309
[Submit][Status][Discuss]
Description
有 n 种数字,第 i 种数字是 ai、有 bi 个,权值是 ci。
若两个数字 ai、aj 满足,ai 是 aj 的倍数,且 ai/aj 是一个质数,
那么这两个数字可以配对,并获得 ci×cj 的价值。
一个数字只能参与一次配对,可以不参与配对。
在获得的价值总和不小于 0 的前提下,求最多进行多少次配对。
Input
第一行一个整数 n。
第二行 n 个整数 a1、a2、……、an。
第三行 n 个整数 b1、b2、……、bn。
第四行 n 个整数 c1、c2、……、cn。
Output
一行一个数,最多进行多少次配对
Sample Input
3
2 4 8
2 200 7
-1 -2 1
2 4 8
2 200 7
-1 -2 1
Sample Output
4
HINT
n≤200,ai≤10^9,bi≤10^5,∣ci∣≤10^5
Source
Solution
二分图建图方式,分析一下
如果$a[i]$和$a[j]$互质,不妨把它们分解质因子,互质即质因子约去后的质因子互质,或者只剩一个质因子
所以考虑先预处理 线性筛质数,再质因数分解,含有奇数个质因子的与含有偶数个质因子的分列两排
如果满足$a[i]$和$a[j]$可连($a[i]$为质因数为奇的$a[j]$为质因数为偶的),则由$i--->j$连容量$inf$,费用$c[i]*c[j]$
由$S--->odd[i]$连容量$b[odd[i]]$,费用为$0$;同理$even[i]--->T$连容量$b[even[i]]$费用$0$
然后跑最大费用最大流,因为最大费用最大流每次增广的路径费用严格不下降,所以,贪心的一直跑到总费用$<=0$的时候,单独判断一下即可
容易出问题的细节:
1.注意开longlong...
2.Inf的范围要注意(习惯性的开0x7fffffff导致WA5组...)
3.不能暴力筛和暴力分解,时间复杂度不科学
Code
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
int read()
{
int x=,f=; char ch=getchar();
while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
return x*f;
}
#define maxn 1010
#define maxm 100010
int n,a[maxn],b[maxn];long long c[maxn]; bool f=;
struct EdgeNode{int next,cap,to,from;long long cost;}edge[maxm];
int head[maxn],cnt=;
void add(int u,int v,int w,long long c)
{
cnt++;
edge[cnt].to=v;edge[cnt].from=u;edge[cnt].cap=w;edge[cnt].cost=c;edge[cnt].next=head[u];head[u]=cnt;
}
void insert(int u,int v,int w,long long c) {add(u,v,w,c); add(v,u,,-c);}
int from[maxn],S,T; long long dis[maxn]; bool mark[maxn];
#define inf 10000000000000LL
bool spfa()
{
queue<int>q;
for (int i=S; i<=T; i++) dis[i]=-inf; memset(from,,sizeof(from));
q.push(S); mark[S]=; dis[S]=;
while (!q.empty())
{
int now=q.front(); q.pop(); mark[now]=;
for (int i=head[now]; i; i=edge[i].next)
if (edge[i].cap && dis[edge[i].to]<dis[now]+edge[i].cost)
{
dis[edge[i].to]=dis[now]+edge[i].cost; from[edge[i].to]=i;
if (!mark[edge[i].to])
q.push(edge[i].to),mark[edge[i].to]=;
}
}
return dis[T]!=-inf;
}
long long Cost; int Flow;
void MaxCostFlow()
{
int flow=inf;
for (int i=from[T]; i; i=from[edge[i].from])
flow=min(flow,edge[i].cap); if ((long long)Cost+flow*dis[T]>=0LL)
{
for (int i=from[T]; i; i=from[edge[i].from])
edge[i].cap-=flow,edge[i^].cap+=flow;
Flow+=flow; Cost+=dis[T]*flow; }
else
Flow+=abs(Cost/dis[T]),Cost-=(Cost/dis[T])*dis[T],f=;
// Flow+=flow;
//printf("%d %I64d %I64d\n",Flow,Cost,dis[T]);
}
int prime[],tot; bool flag[];
void Prework()
{
flag[]=;
for (int i=; i<=; i++)
{
if (!flag[i]) prime[++tot]=i;
for (int j=; j<=tot&&i*prime[j]<=; j++)
flag[i*prime[j]]=;
}
}
int odd[maxn],even[maxn],ot,et;
bool check(int x,int y)
{
if (x%y && y%x || !x || !y) return ;
int tmp=max(x/y,y/x);
for (int i=; i<=tot&&prime[i]<tmp; i++)
if (tmp%prime[i]==) return ;
return ;
}
void Build()
{
Prework();
S=,T=n+;
for (int i=,ct=; i<=n; i++,ct=)
{
for (int j=,tmp=a[i]; j<=tot; j++,tmp=a[i])
while (tmp%prime[j]==) tmp/=prime[j],ct++;
//else if (prime[j]>tmp) break;
if (ct&) odd[++ot]=i; else even[++et]=i;
}
for (int i=; i<=ot; i++)
for (int j=; j<=et; j++)
if (check(a[odd[i]],a[even[j]]))
insert(odd[i],even[j],inf,c[odd[i]]*c[even[j]])/*,printf("%d-->%d:%d/%I64d\n",odd[i],even[j],inf,c[odd[i]]*c[even[j]])*/;
for (int i=; i<=ot; i++) insert(S,odd[i],b[odd[i]],0LL)/*,printf("%d-->%d:%d/%I64d\n",S,odd[i],b[odd[i]],0LL)*/;
for (int i=; i<=et; i++) insert(even[i],T,b[even[i]],0LL)/*,printf("%d-->%d:%d/%I64d\n",even[i],T,b[even[i]],0LL)*/;
//printf("%d %d %d\n",ot,et,cnt);
}
int main()
{
// freopen("menci_pair.in","r",stdin);
// freopen("menci_pair.out","w",stdout);
n=read();
for (int i=; i<=n; i++) a[i]=read();
for (int i=; i<=n; i++) b[i]=read();
for (int i=; i<=n; i++) c[i]=(long long)read();
Build();
while (spfa() && !f) MaxCostFlow();
printf("%d\n",Flow);
return ;
}
当时唯一想出和出题人一模一样的解法的题,但是最后不知道莫名其妙的写错了什么地方,结果连暴力分都不到....(回来之后这么久才想起来重写一遍..)