[BZOJ4857][JSOI2016]反质数序列[最大点独立集]

题意

在长度为 \(n\) 的序列 \(a\) 中选择尽量长的子序列，使得选出子序列中任意两个数的和不为质数。
\(n\leq3000\ ,a_i\leq10^5\).

分析

直接按照奇偶性建立二分图，两个数之和如果为质数连边，跑独立集。

假设\(a+b= p_1\ ,a+c=p_2\) ,在除了 \(1+1=2\) 的情况下 \(b,c\) 奇偶性相同，构成合数。

所以总边数不会达到 \(n^2\) ,注意选出子序列中最多存在一个1.

总时间复杂度为 \(Dinic\) 时间复杂度。

代码

#include<bits/stdc++.h>
using namespace std;
#define go(u) for(int i=head[u],v=e[i].to;i;i=e[i].last,v=e[i].to)
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define pb push_back
#define re(x) memset(x,0,sizeof x)
typedef long long LL;
inline int gi(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
    return x*f;
}
template<typename T>inline bool Max(T &a,T b){return a<b?a=b,1:0;}
template<typename T>inline bool Min(T &a,T b){return b<a?a=b,1:0;}
const int N=3007,inf=0x3f3f3f3f,Num=2e5;
int n;
int vis[200007],a[N],mark[N];
namespace MF{
  int edc=1,S,T,ndc,sum;
  int dep[N],cur[N];
  struct edge{int to,cap,flow;}e[1000007];
  vector<int>G[N];
  void Add(int a,int b,int c){
    e[++edc]=(edge){b,c,0},G[a].push_back(edc);
    e[++edc]=(edge){a,0,0},G[b].push_back(edc);
  }
  bool bfs(){
    queue<int>Q;
    re(dep);re(cur);dep[S]=1,Q.push(S);
    while(!Q.empty()){
      int u=Q.front();Q.pop();
      for(int i=0;i<G[u].size();i++){
        edge &E=e[G[u][i]];
        if(E.cap>E.flow&&!dep[E.to])
            dep[E.to]=dep[u]+1,Q.push(E.to);
      }
    }
    return dep[T];
  }
  int dfs(int u,int b){
    if(u==T||!b) return b;
    int flow=0,f;
    for(int &i=cur[u];i<G[u].size();i++){
      edge &E=e[G[u][i]];
      if(dep[u]+1==dep[E.to]){
        f=dfs(E.to,min(b,E.cap-E.flow));
        flow+=f,E.flow+=f,b-=f,e[G[u][i]^1].flow-=f;
        if(!b) break;
      }
    }
    return flow;
  }
  int maxflow(){
    int res=0;
    while(bfs()) res+=dfs(S,inf);return res;
  }
}
int main(){
#ifndef ONLINE_JUDGE
#endif
    using namespace MF;
    n=gi();S=n+1,T=S+1;int cnt=0;
    for(int i=2;i<=Num;++i) if(!vis[i])
    for(int j=i+i;j<=Num;j+=i) vis[j]=1;
    bool fi=1;
    rep(i,1,n) {
        a[i]=gi();
        if(a[i]==1&&fi) fi=0;
        else if(a[i]==1) mark[i]=1,++cnt;
    }
    rep(i,1,n)if(a[i]&1&&!mark[i])
    rep(j,1,n)if(!mark[j]&&!(a[j]&1)&&!vis[a[i]+a[j]]) Add(i,j,1);
    rep(i,1,n) if(!mark[i]){
        if(a[i]&1) Add(S,i,1);
        else Add(i,T,1);
    }
    printf("%d\n",n-maxflow()-cnt);
    return 0;
}

转载于:https://www.cnblogs.com/yqgAKIOI/p/9794341.html