题意:
给出一定数量的三种颜色的球,计算如何摆放得到值最大(有一定顺序)
有三种摆放方法
1、如果放的是第一个(桌子上原来没有),数值不变
2、如果在末尾追加一个,那么增加前面不同颜色的个数的值
3、如果在两个之间放置,那么增加两边不同颜色的个数的值
思路:
b蓝 r红 y黄
三种摆放方法可以理解为两种,2和3归为一种。
寻求最大值的话,一定是每次在中间放置球,这样得到的值更大
首先,如果每个球大于2个的话,就是说可以拿出部分球摆成以下这样情况
brybry
,两边都有3种颜色,剩下的球全部放在yb
之间,每次增加6
那么只需要计算 (总数-6)* 6 + 摆放6个时得到的分数
若有颜色不到2个,那么一开始两边摆的球的颜色数就会减少
设此时两边的颜色总数和为a,那么后面每个加入的球放中间都能得到a分
即
ans=(总的球数-先摆的球数)*a+b(b为先摆的过程中所获得的得分)
那么b怎么计算呢,在摆放最开始几个球的时候,会发现是一个等差数列,1,2,3...
所以我们直接用等差数列求和就可以得出b的值了
b=(0+先摆的球数-1)*先摆的球数/2
#include<iostream>
using namespace std;
typedef long long ll;
ll deal(ll x) {
return x > 2 ? 2 : x;
}
int main() {
ll a, b, c;
while (~scanf("%lld %lld %lld", &a, &b, &c)) {
ll index = deal(a) + deal(b) + deal(c);
ll sum = a + b + c - index;
ll ans = sum * index + index * (index - 1) / 2;
printf("%lld
", ans);
}
return 0;
}