牛客练习赛114 F-Kevin去砍树

题目的要求是找一个点然后以这个点往左或者往右砍树,要求每砍的高度严格小于上一次砍的树的高度,求权值最大

贪心可以知道以极大值为初始点,往左或者往右砍是最佳的

设前一颗砍的高度为c

设左右边其中一个高度为a另一个为b

如果c>a>b的话因为那么肯定可以先把a砍了再去看后面的

如果a>c>b的话那a就不能砍了,最佳的肯定是把b给砍了

如果c>(a=b)的话我们不知道要砍左边还是右边才是最优的,那我们就可以分两种情况递归下去

当我们左右都砍不了时也就是递归到边界的时候我们就返回

同时在递归的时候保存他们的最大值

cpp 复制代码
const int inf = 0x3f3f3f3f3f3f3f3f, N = 2e5 + 5, mod = 1e9 + 7;
int h[N], w[N];
int n;
int res = 0, maxx = 0;
void cal(int l, int r,int mid)
{
	maxx = max(maxx, res);
	if (h[l] >= h[mid] && h[r] >= h[mid]) return;
	if (l<0 || r>n+1) return;
	if (h[l] > h[r]&&h[mid]>h[l]) {
		res += w[l];
		cal(l - 1, r, l);
		res -= w[l];
	}
	else if (h[l] > h[r] && h[mid] > h[r]) {
		res += w[r];
		cal(l, r + 1, r);
		res -= w[r];
	}
	else if (h[l]<h[r] && h[mid]>h[r]) {
		res += w[r];
		cal(l, r + 1, r);
		res -= w[r];
	}
	else if (h[l]<h[r] && h[mid]>h[l]) {
		res += w[l];
		cal(l - 1, r, l);
		res -= w[l];
	}
	else if (h[l] == h[r]) {
		res += w[l];
		cal(l - 1, r, l);
		res -= w[l];
		res += w[r];
		cal(l, r + 1, r);
		res -= w[r];
	}
}
signed main()
{
	ios_base::sync_with_stdio(0); cin.tie(0), cout.tie(0);
	int T;
	cin >> T;
	while (T--)
	{
		maxx = 0;
		cin >> n;
		for (int i = 1; i <= n; i++) {
			cin >> h[i];
		}
		for (int i = 1; i <= n; i++) {
			cin >> w[i];
			maxx = max(maxx, w[i]);
		}
		h[n + 1] = -1; h[0] = -1;
		w[n + 1] = 0;
		for (int i = 1; i <= n; i++) {
			if (h[i] >= h[i - 1] && h[i] >= h[i + 1]) {
				if (h[i] == h[i - 1] && h[i] == h[i + 1]) continue;
				res = w[i];
				cal(i - 1, i + 1, i);
			}
		}
		cout << maxx << "\n";
	}
}
相关推荐
小辉同志7 小时前
437. 路径总和 III
算法·深度优先·广度优先
笨笨阿库娅7 小时前
从零开始的算法基础学习
学习·算法
不想睡觉_7 小时前
优先队列priority_queue
c++·算法
那个村的李富贵16 小时前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
power 雀儿16 小时前
Scaled Dot-Product Attention 分数计算 C++
算法
琹箐17 小时前
最大堆和最小堆 实现思路
java·开发语言·算法
renhongxia117 小时前
如何基于知识图谱进行故障原因、事故原因推理,需要用到哪些算法
人工智能·深度学习·算法·机器学习·自然语言处理·transformer·知识图谱
坚持就完事了17 小时前
数据结构之树(Java实现)
java·算法
算法备案代理17 小时前
大模型备案与算法备案,企业该如何选择?
人工智能·算法·大模型·算法备案
赛姐在努力.17 小时前
【拓扑排序】-- 算法原理讲解,及实现拓扑排序,附赠热门例题
java·算法·图论