码迷,mamicode.com
首页 > 编程语言 > 详细

【拓扑排序】【bitset】Gym - 101128A - Promotions

时间:2017-08-07 10:19:48      阅读:143      评论:0      收藏:0      [点我收藏+]

标签:ace   front   namespace   div   push   name   using   void   clu   

给你一张DAG,若选择u点,则必须先选择所有能到达其的点。问你在选择A个点的情况下,哪些点必选;选择B个点的情况下,哪些点必选;选择B个点的情况下,哪些点一定不选。

选择A个点的情况,必选的点是那些其所能到达的点数>n-A的点。

选B个点,必选的点与前者类似。

一定不选的点,是能到达它的点数>B的点。

 

处理一个点所能到达的点数,要么暴力,也可以可以将边反向后,对每个点搞个bitset来做。

一个DAG反向后仍是DAG。

#include<cstdio>
#include<bitset>
#include<queue>
using namespace std;
bitset<5010>S[5010];
queue<int>q;
int first[5010],e,next[20010],v[20010],siz[5010],si2[5010];
void AddEdge(int U,int V){
	v[++e]=V;
	next[e]=first[U];
	first[U]=e;
}
int firs2[5010],e2,nex2[20010],v2[20010];
void AddEdg2(int U,int V){
	v2[++e2]=V;
	nex2[e2]=firs2[U];
	firs2[U]=e2;
}
int A,B,n,m,du[5010],du2[5010],ans1,ans2,ans3;
int main(){
	int x,y;
//	freopen("a.in","r",stdin);
	scanf("%d%d%d%d",&A,&B,&n,&m);
	for(int i=1;i<=m;++i){
		scanf("%d%d",&x,&y); ++x; ++y;
		AddEdge(x,y);
		AddEdg2(y,x);
		++du[y];
		++du2[x];
	}
	for(int i=1;i<=n;++i){
		S[i].set(i);
		if(!du2[i]){
			q.push(i);
		}
	}
	while(!q.empty()){
		int U=q.front();
		for(int i=firs2[U];i;i=nex2[i]){
			S[v2[i]]|=S[U];
			--du2[v2[i]];
			if(!du2[v2[i]]){
				q.push(v2[i]);
			}
		}
		q.pop();
	}
	for(int i=1;i<=n;++i){
		siz[i]=S[i].count();
		S[i].reset();
		S[i].set(i);
		if(!du[i]){
			q.push(i);
		}
	}
	while(!q.empty()){
		int U=q.front();
		for(int i=first[U];i;i=next[i]){
			S[v[i]]|=S[U];
			--du[v[i]];
			if(!du[v[i]]){
				q.push(v[i]);
			}
		}
		q.pop();
	}
	for(int i=1;i<=n;++i){
		si2[i]=S[i].count();
	}
	for(int i=1;i<=n;++i){
		if(siz[i]>n-A){
			++ans1;
		}
	}
	for(int i=1;i<=n;++i){
		if(siz[i]>n-B){
			++ans2;
		}
	}
	for(int i=1;i<=n;++i){
		if(si2[i]>B){
			++ans3;
		}
	}
	printf("%d\n%d\n%d\n",ans1,ans2,ans3);
	return 0;
}

【拓扑排序】【bitset】Gym - 101128A - Promotions

标签:ace   front   namespace   div   push   name   using   void   clu   

原文地址:http://www.cnblogs.com/autsky-jadek/p/7296638.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!