HackerRank training-the-army

Posted 北屿

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了HackerRank training-the-army相关的知识,希望对你有一定的参考价值。

Description

有 \(n\) 个技能,每次可以通过一个巫师,将一个技能转化成另一个技能,问最有最多有多少不同的技能.

Sol

网络流.

先说说我一开始非常 naive 的建图,将技能拆点,中间加一列巫师, \(S\) 向初始技能连边容量为个数,对应点之间连边容量为 \(INF\),然后从拆出来的点向 \(T\) 连边,容量为 \(1\) ,巫师从左边连一个点右边连一个点,容量为 \(1\).

然而这样可以过大部分的点...数据好弱...

其实这样建图是错的...我是想着用最大流表示一种情况,但是少考虑了一种情况就是巫师将一个技能换成另一个技能之后还可以继续变换.

这样其实就不用拆点,直接连巫师,然后从巫师直接连回去就可以了...

Code

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<iostream>
using namespace std;

const int N = 555;

inline int in(int x=0,char ch=getchar()){ while(ch>‘9‘ || ch<‘0‘) ch=getchar();
	while(ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();return x; }

struct NetWork{
	struct Edge{ int fr,to,flow; };
	vector<Edge> edge;
	vector<int> g[N];
	int p[N],cur[N],d[N],a[N],aa[N],w[N];
	int s,t,m,n,k,flow,cp;
	
	void Add_Edge(int fr,int to,int fl){
		edge.push_back((Edge){ fr,to,fl }),edge.push_back((Edge){ to,fr,0 });
		m=edge.size(),g[fr].push_back(m-2),g[to].push_back(m-1);
	}
	int BFS(){
		memset(d,0,sizeof(d));d[s]=1;
		queue<int> q;q.push(s);
		for(int x;!q.empty();){
			x=q.front(),q.pop();
			for(int i=0,v;i<g[x].size();i++) if(!d[v=edge[g[x][i]].to] && edge[g[x][i]].flow>0)
				d[v]=d[x]+1,q.push(v);
		}return d[t]>0;
	}
	int Dinic(){
		flow=0;
		for(int x,k,mine,minf;BFS();){
			for(memset(cur,0,sizeof(cur)),k=0,x=s;;){
				if(x==t){
					mine=-1,minf=0x7fffffff;
					for(int i=0;i<k;i++) if(edge[p[i]].flow < minf) minf=edge[p[i]].flow,mine=i;
					for(int i=0;i<k;i++) edge[p[i]].flow-=minf,edge[p[i]^1].flow+=minf;
					k=mine,flow+=minf,x=edge[p[mine]].fr;
				}
				for(int &i=cur[x];i<g[x].size();i++){
					Edge &e=edge[g[x][i]];
					if(e.flow>0 && d[x]+1==d[e.to]) break;
				}
				if(cur[x]<g[x].size()){
					p[k]=g[x][cur[x]],x=edge[p[k++]].to;
				}else{
					if(!k) break;
					d[x]=-1,x=edge[p[--k]].fr;
				}
			}
		}return flow;
	}
	void init(){
		n=in(),k=in();
		s=n+k+1,t=s+1;
		for(int i=1,x;i<=n;i++) x=in(),Add_Edge(s,i,x),Add_Edge(i,t,1);
		for(int i=1,x,y;i<=k;i++){
			x=in();
			for(int j=1,tmp;j<=x;j++) tmp=in(),Add_Edge(tmp,n+i,1);
			y=in();
			for(int j=1,tmp;j<=y;j++) tmp=in(),Add_Edge(n+i,tmp,1);
		}
		cout<<Dinic()<<endl;
	}
}sol;

int main(){
	sol.init();
	return 0;
}

  

以上是关于HackerRank training-the-army的主要内容,如果未能解决你的问题,请参考以下文章

不可分子集 - Hackerrank

堆栈和优化——来自hackerrank的例子

Hackerrank:Sherlock 和 Anagrams [关闭]

如何清除 Hackerrank 动态数组运行时错误?

HackerRank 天气观测站 5

没有指针的 C++ 分段错误 (HackerRank)