题目链接
https://codeforces.com/contest/808/problem/F
题意
给出n个三元组<p,c,l>,任选若干个,要求符合
- c相加为质数的两个组不能同时选
- ∑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;
}