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

题目链接

(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
",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
",ans[i].first,ans[i].second);

	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/9724789.html