标签:贪心
描述:
小蓝正在玩一款游戏。
游戏中魏蜀吴三个国家各自拥有一定数量的士兵 X,Y,Z(一开始可以认为都为 0)。
游戏有 n 个可能会发生的事件,每个事件之间相互独立且最多只会发生一次,当第 i 个事件发生时会分别让 X,Y,Z 增加 Ai,Bi,Ci。
当游戏结束时 (所有事件的发生与否已经确定),如果 X,Y, Z 的其中一个大于另外两个之和,我们认为其获胜。
例如,当 X>Y+Z 时,我们认为魏国获胜。
小蓝想知道游戏结束时如果有其中一个国家获胜,最多发生了多少个事件?
如果不存在任何能让某国获胜的情况,请输出 −1
。
输入格式
输入的第一行包含一个整数 n。
第二行包含 n 个整数表示 Ai,相邻整数之间使用一个空格分隔。
第三行包含 n 个整数表示 Bi,相邻整数之间使用一个空格分隔。
第四行包含 n 个整数表示 Ci,相邻整数之间使用一个空格分隔。
输出格式
输出一行包含一个整数表示答案。
数据范围
对于 40% 的评测用例,n≤500;
对于 70% 的评测用例,n≤5000;
对于所有评测用例,1≤n≤10^5,0≤Ai,Bi,Ci≤10^9。
注意,蓝桥杯官方给出的关于 Ai,Bi,Ci 的数据范围是 1≤Ai,Bi,Ci≤10^9,但是这与给出的输入样例相矛盾,因此予以纠正。
输入样例:
3
1 2 2
2 3 2
1 0 7
输出样例:
2
样例解释
发生两个事件时,有两种不同的情况会出现获胜方。
发生 1,2 事件时蜀国获胜。
发生 1,3 事件时吴国获胜。
讲讲思路:
既然要让国家获胜,那一定要选对该国有利的事件.
我们可以对每个事件计算对该国的贡献,例如1 5 3 对三个国家的贡献分别就为 -7 1 -3
那么在判断国家能否胜利就可以先降序排序,然后先加正的贡献,后加负的贡献,直到比0小
最后判断一下哪个国家的事件是最多的即可.
以下是AC代码:
#include <iostream>
#include <cstring>
#include <algorithm>using namespace std;
/* 定义方便 */
typedef long long LL;
const int N = 1e5 + 10;/* 定义变量 */
int n;
LL a = 0, b = 0, c = 0;
LL A[N] , B[N] , C[N];/* 快排降序 */
bool cmp(int a ,int b)
{return a > b;
}/* 三种情况各个判断 */
LL win(int x)
{LL cnt = 0 , sum = 0; // cnt计数,sum记总和/* 拿其中一种分析 */if(x == 1){sort(A , A + n , cmp); // 先排序降序for (int i = 0; i < n; i ++ ) // 将贡献大的先加上,直到sum小于0前的前一个停止if(sum + A[i] > 0) sum += A[i] , cnt ++;}if(x == 2){sort(B , B + n , cmp);for (int i = 0; i < n; i ++ )if(sum + B[i] > 0) sum += B[i] , cnt ++;}if(x == 3){sort(C , C + n , cmp);for (int i = 0; i < n; i ++ )if(sum + C[i] > 0) sum += C[i] , cnt ++;}if(cnt == 0) cnt = -1; // 如果cnt不变,改为-1return cnt;
}int main()
{/* 正常读入 */cin >> n;for (int i = 0; i < n; i ++ )cin >> A[i];for (int i = 0; i < n; i ++ )cin >> B[i];for (int i = 0; i < n; i ++ )cin >> C[i];/* 核心 *//* 计算各个序列贡献 */for (int i = 0; i < n; i ++ ){LL a1 = A[i], b1 = B[i], c1 = C[i]; // 单独拿变量赋值,防止变动后连锁反应A[i] = a1 - b1 - c1;B[i] = b1 - a1 - c1;C[i] = c1 - b1 - a1;}/* 比较ans */LL ans = -1;for (int i = 1; i <= 3; i ++ )ans = max(win(i) , ans); // 三种情况取最大cout << ans;return 0;
}