Codeforces Round #660 (Div. 2) D - Captain Flint and Treasure

题意: 编号为 [ 1 , n ] [1,n] [1,n]的点初始权值为 a i a_i ai,初始答案为 0 0 0,每次选取一个点 i i i后,答案加上 a i a_i ai,并且 i i i点不可再次被选,且如果 a i ≠ 1 a_i\neq1 ai=1,给 a b i a_{b_i} abi加上 a i a_i ai。问可以得到的最大答案和该答案上选择点的顺序。
题解: 典型的拓扑排序,先对每个点计度数。
从度为 0 0 0的点开始选择:

  • 如果点 i i i当前权值小于 0 0 0,选择点 i i i会减少 a b i a_{b_i} abi,所以暂时不选(形式即为给 a b i a_{b_i} abi加上 0 0 0)。
  • 如果点 i i i的权值不小于 0 0 0,选择该点不会使 a b i a_{b_i} abi减少,故选择该点。
  • 每次枚举完一个点 i i i后,将 b i b_i bi度数相应减 1 1 1

具体实现时可以按照拓扑排序直接写即可。
在输出选择顺序时,先从原始度数最小的点开始,将所有 a i a_i ai大于等于 0 0 0的点输出,因为越在前面对后面的点就有影响。然后从原始度数最大的点开始,将所有 a i a_i ai小于 0 0 0的点输出,因为越在后面的点先选择,就不会对其后面的点产生负影响。

代码:

#include
using namespace std;

typedef long long ll;

template<typename T>
inline T Read(){
    T s = 0, f = 1; char ch = getchar();
    while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
    while(isdigit(ch)) {s = (s << 3) + (s << 1) + ch - '0'; ch = getchar();}
    return s * f;
}

#define read() Read()
#define readl() Read()
const int N = 2e5 + 10;

int n, b[N], d[N];
ll a[N], res;

int p[N], cnt;
int q[N];
void topsort() {
	int hh = 0, tt = -1;
	for(int i = 1; i <= n; ++i) 
		if(!d[i]) q[++tt] = i;
	
	while(hh <= tt) {
		int u = q[hh++];
		p[++cnt] = u;
		res += a[u];
		if(~b[u]) {
			--d[b[u]];
			a[b[u]] += max(0ll, a[u]);
			if(!d[b[u]]) q[++tt] = b[u];
		}
	}
}

void solve() {
	n = read();
	for(int i = 1; i <= n; ++i) d[i] = 0;
	for(int i = 1; i <= n; ++i) a[i] = readl();
	for(int i = 1; i <= n; ++i) {
		b[i] = read();
		if(b[i] == -1) continue;
		++d[b[i]];
	}
	
	topsort();
	
	printf("%lld\n", res);
	vector<int> ans;
	
	for(int i = 1; i <= n; ++i) if(a[p[i]] >= 0) ans.push_back(p[i]);
	for(int i = n; i >= 1; --i) if(a[p[i]] < 0) ans.push_back(p[i]);
	for(int i = 0; i < n; ++i) printf("%d%c", ans[i], " \n"[i == n - 1]);
}

int main()
{
	int T = 1;
	//T = read();
	for(int i = 1; i <= T; ++i) {
		solve();
	}
}

你可能感兴趣的:(Codeforces)