【蓝桥杯每日一题】3.13 陨石

一道简单的模拟题

原题链接：B3938陨石 - 洛谷

大致题意：

• $n$ 间牛棚，防御值$a_i$
• $t$秒后，陨石雨来临，$m$块陨石撞击牛棚，第$i$块撞击第$x_i$间牛棚
• 每撞一次，牛棚防御力-2 防御力$<=0$时，牛棚被破坏
• 有补给，每个补给给牛棚增加$1$点防御值，每次补给需要$1$秒时间
• 求：最多能保护多少个牛棚？

解题思路：（先砸后补救）

• 对陨石砸向的牛棚进行防御力的扣除
• 由于要求保证牛棚最后存在的数量最多，所以要先找出需要补给最少量的那个牛棚，故按照牛棚的防御力进行降序排序，遍历数组，可以补给至“复活”的牛棚数量记为$len1$，防御力$>0$的牛棚数量记为$len2$
• 最后保护的牛棚数量即为$len1+len2$

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int T;
int n,t,m; 
int a[N],x[N];

bool cmp(int a,int b){
	return a>b;
}

int main(){
	scanf("%d",&T);
	
	while(T--)
	{
		scanf("%d%d%d",&n,&t,&m);
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
		}
		for(int i=0;i<m;i++)
		{
			scanf("%d",&x[i]);
			a[x[i]-1]-=2;//根据陨石去掉牛棚的防御值
		}
	
		//对牛棚防御值排序，优先补齐防御值<=0且所需补给最少的
		//目的是尽可能地多挽救牛棚
		sort(a,a+n,cmp); //降序排序
		int ans1=0; //被保护的牛棚数量
		int ans2=0; // 不用保护的牛棚数量
		for(int i=0;i<n;i++)
		{
			//printf("%d ",a[i]);
			if(a[i]<=0&&t+a[i]>0)
			{
				t=t-(1-a[i]);
				ans1++;
			}
			if(a[i]>0)
			{
				ans2++;
			}
		}
		printf("%d\n",ans1+ans2);
	}
	
	return 0;
}