首页 > 代码库 > UVALive 6432 Influence 搜索 剪枝大法好
UVALive 6432 Influence 搜索 剪枝大法好
有n个人,有k个人可以选作传播疾病的母体,和病人直接接触的未被感染者会被感染,求出选择k个人中的哪个可以取得最多的病人数目,有相同的取编号小的那个。
简单搜索,剪枝是如果一个同为母体的可以被其他母体直接或间接传染,这个母体就肯定不会是最多的那个,只会是一条分支。
#include<cstring> #include<string> #include<fstream> #include<iostream> #include<iomanip> #include<cstdio> #include<cctype> #include<algorithm> #include<queue> #include<map> #include<set> #include<vector> #include<stack> #include<ctime> #include<cstdlib> #include<functional> #include<cmath> using namespace std; #define PI acos(-1.0) #define MAXN 10100 #define eps 1e-7 #define INF 0x7FFFFFFF #define LLINF 0x7FFFFFFFFFFFFFFF #define seed 131 #define MOD 1000000007 #define ll long long #define ull unsigned ll #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 struct node{ int v,next; }edge[250000]; int head[5010],a[5010],vis[5010]; int num[5100]; int cnt,ans,maxm,sum; void add_edge(int u,int v){ edge[cnt].v = v; edge[cnt].next = head[u]; head[u] = cnt++; } int dfs(int u){ int i,j; int sum=1; vis[u]=1; for(i=head[u];i!=-1;i=edge[i].next) { int v=edge[i].v; if(num[v]) { num[v]=0; } if(!vis[v]) { sum+=dfs(v); } } return sum; } int main(){ char c; int n,i,j,x,t,m; while(scanf("%d%d",&n,&m)!=EOF) { memset(num,0,sizeof(num)); memset(head,-1,sizeof(head)); cnt = 0; maxm=0; for(i=0;i<m;i++) { scanf("%d",&a[i]); num[a[i]]=1; } for(i=0;i<n;i++) { scanf("%d",&x); while(1) { c = getchar(); if(c=='\n') break; scanf("%d",&t); add_edge(x,t); } } for(i=0;i<m;i++) { if(num[a[i]]) { for(j=1;j<=n;j++) { vis[j] = 0; } int sum=dfs(a[i]); if(sum>maxm){ maxm = sum; ans = a[i]; } } } printf("%d\n",ans); } return 0; }
UVALive 6432 Influence 搜索 剪枝大法好
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。