Codeforces 808F 最大权独立点集,二分

题目链接

https://codeforces.com/contest/808/problem/F

题意

给出n个三元组<p,c,l>,任选若干个,要求符合

  1. c相加为质数的两个组不能同时选
  2. ∑p>=k

求所有可行选择方法中max{l}最小值

思路

首先可知答案单调,我们可以二分l来进行处理

至于每次处理,条件一中相加为质数的情况其实可以分为两种:1+1和奇数加偶数。对于1+1型,显然我们只能选择一个p最大的1时是最优的,而奇数+偶数这个条件就很有意思了,我们把c按奇偶性划分,对所有不能一起选(c相加为质数)的点连边,可以得到一张二分图,而这道题其实就是求二分图最大权独立点集。

我们知道二分图最大权独立点集就设置源点汇点,分别连边权为p,跑出最大流后用∑q-maxflow即为答案。

因此我们二分l,对于每次二分重新建图,只留下l小于本次二分限制的点(c为1的点只留p最大的),跑dinic即可

复杂度

O ( 能 过 ) O(能过) O()

代码
#include<cstdio>
#include<iostream>
#include<iomanip>
#include<map>
#include<unordered_map>
#include<string>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib> 
#include<chrono>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
//#define int long long
//#define double long double
using namespace std;
	typedef long long ll;
	const int maxn=2505;
    const int maxe=200505;
	const int inf=0x3f3f3f3f;
	int n,m,k;
	int head[maxn],cnt;
	struct Edge{
		int v;
		int w;
		int next;
	}edge[maxe];
	int deep[maxn];
	int now[maxe];
    int prime[maxe];
	bool vis[maxe];
	int p[maxn],c[maxn],lv[maxn];
	void prme(){
        vis[1]=1;
		for(int i=2;i*i<maxe;i++){
            if(!vis[i])
			    for(int j=i+i;j<maxe;j+=i)
				    vis[j]=1;
		}
	} 

	void init(){
		memset(head,-1,sizeof(head));
		cnt=0;
		return ;	
	}
	inline void add(int u,int v,int w){
  		//cout<<u<<' '<<v<<' '<<w<<endl;
          edge[cnt].v=v;
		edge[cnt].w=w;
		edge[cnt].next=head[u];
		head[u]=cnt++;
	}
	
	inline void addd(int u,int v,int w){
		//cout<<u<<" "<<v<<" "<<w<<endl; 
		add(u,v,w);
		add(v,u,0);
	}
	
	inline bool bfs(int s,int t){
    	memset(deep,0x3f,sizeof(deep));
    	queue<int>q;
    	q.push(s);deep[s] = 0;now[s] = head[s];
    	while(q.size()){
        	int x = q.front();q.pop();
        	for(int i=head[x];~i;i=edge[i].next){
        	    int y=edge[i].v;
         	    if(edge[i].w>0&&deep[y]==inf){
         	    	q.push(y);
        	        now[y]=head[y];
         	       	deep[y]=deep[x]+1;
        	        if(y==t)	return 1;
				}
        	}
    	}
    	return 0;
	}

	ll dfs(int x,int t,int flow){
    	if(x==t)	return flow;
    	ll ans = 0,k,i;
    	for(i=now[x];~i&&flow;i=edge[i].next){
        	now[x]=i;
        	int y=edge[i].v;
        	if(edge[i].w>0&&(deep[y]==deep[x]+1)){
        	    k=dfs(y,t,min(flow,edge[i].w));
         		if(!k)	deep[y]=inf;
            	edge[i].w-=k;
            	edge[i^1].w+=k;
            	ans+=k;
            	flow-=k;
        	}
    	}
    	return ans;
	}	

	ll dinic(int s,int t){
		ll maxflow=0;
    	while(bfs(s,t))
    	    maxflow+=dfs(s,t,inf);
    	return maxflow;
	}
    bool judge(int lvv){
        init();
        int id=0;p[0]=-1;
        for(int i=1;i<=n;i++){
            if(lv[i]>lvv)   continue;
            if(c[i]==1){
                if(p[i]>p[id])  id=i;
            }
        }
        int tot=0;
        int s=maxn-2,t=s+1;
        for(int i=1;i<=n;i++){
            if(c[i]==1&&id!=i)  continue;
            if(lv[i]>lvv)   continue;
            tot+=p[i];
            if(c[i]&1){
                addd(s,i,p[i]);
            }
            else{
                addd(i,t,p[i]);
            }
        }
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                if(lv[i]<=lvv&&lv[j]<=lvv&&(c[i]!=1||i==id)&&(c[j]!=1||j==id)){
                    if(!vis[c[i]+c[j]]){
                        if(c[i]&1){
                            add(i,j,inf);
                            add(j,i,0);
                        }
                        else{
                            add(j,i,inf);
                            add(i,j,0);
                        }
                    }
                }
            }
        }
        int rec=tot-dinic(s,t);
        return rec>=m?1:0;
    }
    signed main(){
		IOS
		#ifndef ONLINE_JUDGE
		    freopen("IO\\in.txt","r",stdin);
		    freopen("IO\\out.txt","w",stdout);
        #endif
		prme();
        cin>>n>>m;
        int l=-1,r=-1,mid;
        for(int i=1;i<=n;i++){
            cin>>p[i]>>c[i]>>lv[i];
            r=max(r,lv[i]);
        }
        r+=10;
        int ans=-1;
        while(l<=r){
            mid=(l+r)>>1;
            
            if(judge(mid)){
                ans=mid;
                r=mid-1;
            }
            
            else{
                l=mid+1;
            }
            //cout<<mid<<endl<<endl<<endl;
        }
        cout<<ans<<endl;
	} 
						
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值