Codeforces.1045A.Last chance(最大流ISAP 线段树优化建图)

时间:2021-05-25 19:33:03

题目链接

\(Description\)

你需要用给定的\(n\)个武器摧毁\(m\)架飞船中的某一些。每架飞船需要被摧毁恰好一次。

武器共三种:1.可以在给定的集合中摧毁一架飞船;2.可以摧毁区间\([l,r]\)中的一架飞船;3.摧毁给定三架飞船中的恰好两架(保证武器3中给定的所有飞船不同)。

某些武器可以不用。求最多能摧毁飞船的数量。

\(Solution\)

显然网络流。武器1给定的总集合大小有限制,可以直接连边。武器2需要对区间连边,显然线段树优化。

至于武器3,恰好2架的限制可以先不管,对三个点连容量为2的边。如果最后它只匹配了一个,另两个肯定是被武器1或2匹配,给它即可。

至于输出方案,只保留原图中正向的边,其可走次数为反向弧的流量。

然后对每条流DFS,每次找最靠近源点和汇点的两个点就行了(找完把可走次数减掉)。

最后枚举武器3,分配一波。

//62ms	9800KB
#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>
#define gc() getchar()
#define mp std::make_pair
#define pr std::pair<int,int>
const int S=5005*2,N=5005*3,M=(150000+100000)*2,INF=0x3f3f3f3f; int n,tot,Fr,To,A[5005],B[5005],C[5005],cnt[5005],son[S][2];
int src,des,Enum,cur[N],H[N],nxt[M],to[M],cap[M],fr[M],pre[N],num[N],lev[N];
std::vector<pr> ans,e[N]; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
inline void AE(int u,int v,int w)
{
to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum, cap[Enum]=0, fr[Enum]=v;
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum, cap[Enum]=w, fr[Enum]=u;//odd:u->v
}
//-----Segment Tree-----
#define ls son[x][0]
#define rs son[x][1]
#define lson son[x][0],l,m
#define rson son[x][1],m+1,r
void Build(int &x,int l,int r)
{
if(l==r) {x=l; return;}
x=++tot;
int m=l+r>>1;
Build(lson), Build(rson);
AE(x,ls,INF), AE(x,rs,INF);
}
void Insert(int x,int l,int r,int L,int R,int s)
{
if(L<=l && r<=R) {AE(s,x,1); return;}
int m=l+r>>1;
if(L<=m) Insert(lson,L,R,s);
if(m<R) Insert(rson,L,R,s);
}
//-----ISAP-----
bool BFS()
{
static int q[N];
for(int i=0; i<=tot; ++i) lev[i]=tot+1;
int h=0,t=1; lev[des]=0, q[0]=des;
while(h<t)
{
int x=q[h++];
for(int i=H[x]; i; i=nxt[i])
if(lev[to[i]]==tot+1 && cap[i^1])
lev[to[i]]=lev[x]+1, q[t++]=to[i];
}
return lev[src]<=tot;
}
inline int Augment()
{
for(int i=des; i/*!=src*/; i=fr[pre[i]])//i
--cap[pre[i]], ++cap[pre[i]^1];
return 1;
}
int ISAP()
{
if(!BFS()) return 0;
for(int i=0; i<=tot; ++i) ++num[lev[i]],cur[i]=H[i];
int res=0,x=0;//src
while(lev[src]<=tot)
{
if(x==des) res+=Augment(),x=0;//src
bool can=0;
for(int i=cur[x]; i; i=nxt[i])
if(lev[to[i]]==lev[x]-1 && cap[i])
{
can=1, cur[x]=i, pre[x=to[i]]=i;
break;
}
if(!can)
{
int mn=tot;
for(int i=H[x]; i; i=nxt[i])
if(cap[i]) mn=std::min(mn,lev[to[i]]);
if(!--num[lev[x]]) break;
++num[lev[x]=mn+1], cur[x]=H[x];
if(x/*!=src*/) x=fr[pre[x]];
}
}
return res;
}
//-----DFS for Ans-----
bool DFS(int x)
{
if(x==des) return 1;
for(int i=0,l=e[x].size(); i<l; ++i)
// if(i&1 && cap[i^1])//把边重建常数小很多
if(e[x][i].second)
{
if(!DFS(e[x][i].first)) continue;
--e[x][i].second;
if(!x/*==src*/) Fr=e[x][i].first;
if(e[x][i].first==des) To=x;
return 1;
}
return 0;
}
//-----Main-----
int main()
{
#define S root,1,m
n=read(); int m=read(),root=0;//n:weapons m:ships
Enum=1, tot=m, Build(S), src=0, des=++tot;
for(int i=1; i<=m; ++i) AE(i,des,1); for(int i=1,p=tot+1,opt,l,r; i<=n; ++i,++p)
if(!(opt=read()))
{
AE(0,p,1);
for(int k=read(); k--; AE(p,read(),1));
}
else if(opt==1)
l=read(), r=read(), AE(0,p,1), Insert(S,l,r,p);
else
AE(0,p,2), AE(p,A[i]=read(),1), AE(p,B[i]=read(),1), AE(p,C[i]=read(),1);
int tmp=tot; tot+=n;
int flow=ISAP();
printf("%d\n",flow); for(int i=2; i<=Enum; i+=2) e[to[i]].push_back(mp(fr[i],cap[i]));
for(int i=1; i<=flow; ++i)//weapon -> ship
Fr=0, DFS(0), ans.push_back(mp(Fr-tmp,To)), ++cnt[Fr-tmp];
int size=ans.size();//flow
for(int i=1; i<=n; ++i)
if(A[i] && cnt[i]==1)
for(int j=0; j<size; ++j)
if(ans[j].first!=i&&(ans[j].second==A[i]||ans[j].second==B[i]||ans[j].second==C[i]))
{
ans[j].first=i;
break;
}
for(int i=0; i<size; ++i) printf("%d %d\n",ans[i].first,ans[i].second); return 0;
}