力扣508周赛

第一题 3974. K 个元素的最大总和

• 题解

按从大到小依次选k个元素，判断mul的正负决定累加时是否乘以mul，mul减一。
其中，数组长度为10^5所以每次去最大元素用暴暴力遍历会超，我这里用大跟堆实现。下面给出小跟堆代码。

void up(int i, vector<int> & a) { int t = i >> 1; if (!t) return; if (a[t] < a[i]) { swap(a[t], a[i]); up(t, a); } } void down(int i, int n, vector<int>& a) { int l = i * 2; int r = i * 2 + 1; if (r < n) { int k = r; if (a[l] > a[r]) { k = l; } if (a[i] < a[k]) { swap(a[i],a[k]); down(k, n, a); } } }else if (l < n) { if (a[i] < a[l]) { swap(a[i], a[l]); down(l, n, a); } } }

第二题