题解 | #牛牛打怪兽#
牛牛打怪兽
http://www.nowcoder.com/practice/a3b055dd672245a3a6e2f759c237e449
思路:
根据题目要求打第X怪兽的时候,同时会打到第2X、2X+1这两个怪兽,组合拳必须攻击三只怪兽,可以得到以下结论:
怪兽数量至少3只、怪兽数量是奇数。
那么要想打到最后一只怪兽(A[n-1]),就得对A[n/2-3/2]打拳,这样可以保证满足题目要求。
于是我们可以从A[n/2-3/2]开始,往前遍历,对A[n/2-3/2]的打拳数量应该是A[n-1]和A[n-2]中较大数的值。这样才能保证把两个都打死。(注:下标的倍数关系是打A[i]的同时会打到A[2i+1],A[2i+2])
所以可以用一个for循环,i从n/2-1开始递减,直到i=0停止。A[i]被打的次数(tmp)就是A[2i+1]和A[2i+2]中较大的数,然后更新A[i]的值(这里的更新是指A[i]有可能没被打死,那么就要比较0和A[i]-tmp的大小),res累加tmp即可,最后返回值的时候只要加上A[0],也就是第1只怪兽的剩余血量。
代码如下
class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param n int整型 n个怪兽 * @param A int整型vector A数组代表每个怪兽的血量 * @return int整型 */ int slove(int n, vector<int>& A) { // write code here int res = 0; if((n<=2) || !(n%2)) return -1;//必须打三只以上怪兽,n必须是奇数 for(int i=n/2-3/2;i>=0;i--) { int tmp = max(A[2*i+1],A[2*i+2]); A[i] =max(0, A[i]-tmp); res += tmp; } return res+A[0]; } };
牛客刷题记录 文章被收录于专栏
记录自己的刷题记录,刷过的题的解法