【SSL 2876】工程【拓扑排序+DP】

2017年东莞市信息学特长生测试题 工程

Time Limit:10000MS Memory Limit:256000K
Case Time Limit:1000MS

Description

张三是某工程公司的项目工程师。一天公司接下一项大型工程，该公司在大型工程的施工前，先要把整个工程划分为若干个子工程，并把这些子工程编号为1、2、…、N；这样划分之后，子工程之间就会有一些依赖关系，即一些子工程必须在某些子工程完成之后才能施工，公司需要工程师张三计算整个工程最少的完成时间。
对于上面问题，可以假设：
1、根据预算，每一个子工程都有一个完成时间。
2、子工程之间的依赖关系是：部分子工程必须在一些子工程完成之后才开工。
3、只要满足子工程间的依赖关系，在任何时刻可以有任何多个子工程同时在施工，也即同时施工的子工程个数不受限制。
例如：有五个子工程的工程规划表：

现在对于给定的子工程规划情况，及每个子工程完成所需的时间，如果子工程划分合理则求出完成整个工程最少要用的时间，如果子工程划分不合理，则输出-1。

Input

第1行为正整数N，表示子工程的个数（N<=200）
第2行为N个正整数，分别代表子工程1、2、…、N的完成时间。
第3行到N+2行，每行有N-1个0或1，其中的第K+2行的这些0或1，分别表示“子工程K”与子工程1、2、…、K-1、K+1、…、N的依赖关系（K=1、2、…、N）。每行数据之间均用空格分开。

Output

如果子工程划分合理则输出完成整个工程最少要用的时间，如果子工程划分不合理，则输出-1。

Sample Input

样例1：

5
5 4 12 7 2
0 0 0 0
0 0 0 0
0 0 0 0
1 1 0 0
1 1 1 1

样例2：

5
5 4 12 7 2
0 1 0 0
0 0 0 0
0 0 1 0
1 1 0 0
1 1 1 1

Sample Output

样例1：

14

样例2：

-1

分析：

这道题和奖金其实差不多
有几点不同：
奖金要求的是总和 而这道题要求的是单个的时间
同样 奖金的$DP$是父节点$+1$ 这道题直接父节点与自己比较就可以了
另外 这道题要多开一个数组 存工作量
微改的$DP$方程：
$$f[a[i].to]=max(f[a[i].to],f[que[head]])$$

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,x,tot,h[10005],f[10005],in[10005],v[10005],que[10005],p[10005],head,tail,ans; 
struct node{
	int to,next;
}a[10005];
void add(int x,int y){
	a[++tot]=(node){y,h[x]};  //邻接表
	h[x]=tot;
}
void topsort(){  //拓扑排序
	for(int i=1;i<=n;i++)
	{
		if(!in[i])
		{
			f[i]=p[i];  //存工作量
			que[++tail]=i;  //入队
			v[i]=1;  //标记
		}
	}
	while(head<tail)
	{
		head++;
		for(int i=h[que[head]];i;i=a[i].next)
		{
			if(!v[a[i].to])
			{
				f[a[i].to]=max(f[a[i].to],f[que[head]]);  //父节点与自己比较
				in[a[i].to]--;  //删边
				if(!in[a[i].to])  //入度为0
				{
					f[a[i].to]+=p[a[i].to];  //累计工作量
					v[a[i].to]=1;  //同上
					que[++tail]=a[i].to;
				}
			}
		}
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&p[i]);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	{
		if(i!=j)
		{
			scanf("%d",&x);
			if(x)
			{
				add(j,i);
				in[i]++;  //统计入度
			}
		} 
	}
	topsort();
	for(int i=1;i<=n;i++)
	{
		if(in[i]!=0) {  //存在环
			printf("-1");
			return 0;
		}else{
			ans=max(f[i],ans);  //找最优
		}
	}
	printf("%d",ans);
	return 0;
}