d森林问题——贪心算法

问题

编程实现 T/S 的 d 森林问题。 设 T 为一带权树，树中的每个边的权都为整数。
又设 S 为 T 的一个顶点的子集，从 T 中删除 S 中的所有结点，则得到一个森林，记
为 T/S。如果 T/S 中所有树从根到叶子节点的路径长 度都不超过 d，则称 T/S 是一个
d 森林。设计一个算法求 T 的最小顶点集合 S， 使 T/S 为一个 d 森林

思路

代码

#include <iostream>
using namespace std;

typedef struct  node {
	int father;   //父节点
	int distance;   //到父节点的路长
	int out;        //出度
	int maxroad;    //到叶节点的路畅
	int cut;       //切掉标志
} node, *pnode;

class dTree {
	private:
		pnode tree;
		int BiggestDistance, numberofnode, sum, qhead, qend;
		int que[1000];
	public:
		dTree(int n, int d) {        //构建与初始化树
			sum = 0;
			qhead = 0;
			qend = 0;
			BiggestDistance = d;
			numberofnode = n;
			tree = new node[n];
			tree[0].father = -1;
			tree[0].distance = 0;
			for (int i = 0; i < n; i++) {
				int NumOfSon, WeightOfRoad, num;
				cin >> NumOfSon ;
				tree[i].cut = 0;
				tree[i].out = NumOfSon ;
				tree[i].maxroad = 0;
				for (int j = 0; j < NumOfSon; j++) {
					cin >>  num >> WeightOfRoad ;
					tree[num].father = i;
					tree[num].distance =  WeightOfRoad;
				}
			}
		}
		void push(int i) {              //入队列
			que[qend] = i;
			qend++;
		}
		int pop() {                  //出队列
			int temp = que[qhead++];
			return temp;
		}
		int cuttree() {                   //剪切
			for (int i = numberofnode - 1; i >= 0; i--) {
				if (tree[i].out == 0) {
					push(i);
				}
			}
			while (qhead != qend) {
				int temp = pop();
				int len = tree[temp].distance;
				int par = tree[temp].father;
				if (tree[par].cut == 0 && tree[temp].maxroad + len > BiggestDistance) {
					tree[par].cut = 1;
					par = tree[par].father;
					sum++;
				} else if (tree[temp].cut == 0 && tree[par].maxroad < tree[temp].maxroad + len) {
					tree[par].maxroad = tree[temp].maxroad + len;
				}
				if (--tree[par].out == 0)
					push(par);
			}
			return sum;
		}
};

int main() {
	int n, d;
	cin >> n >> d;
	dTree test(n, d);
	cout << test.cuttree();
	return 0;
}