hdu 4811 Ball 贪心

问题相当于把尽可能多的颜色放到左右两边

而只有3种颜色

易知两轮的分配之后，不管往中间加入什么颜色，最后都是增加一个固定值，记做ccnt

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <vector>
#include <set>
#include <queue>
#include <map>
using namespace std;
#define INF 1e9
#define maxn
#define rep(i,x,y) for(int i=x;i<=y;i++)
#define mset(x) memset(x,0,sizeof(x))
typedef __int64 ll;


int a[4];
int ans[3][3][3];
int main(){
//	freopen("a.txt","r",stdin);
//	freopen(".out","w",stdout);
	ans[0][0][0]=0;
	ans[0][0][1]=0;
	ans[0][1][1]=1;
	ans[1][1][1]=3;
	ans[0][0][2]=1;
	ans[0][1][2]=3;
	ans[1][1][2]=6;
	ans[0][2][2]=6;
	ans[1][2][2]=10;
	ans[2][2][2]=15;
	while(cin>>a[1]>>a[2]>>a[3]){
		sort(a+1, a+4);
		int ccnt=0;
		//两轮的分配之后还剩下的数目
		int num=0;
		rep(i,1,3){
			if(a[i]>2){//把i颜色分配到两边
				ccnt += 2;
				num += a[i]-2;
				a[i] = 2;
			}
			else{
				ccnt += a[i]; //如果是1的话把i颜色分配到其中一边
			}
		}
		printf("%I64d\n", (ll)ans[a[1]][a[2]][a[3]] + (ll)ccnt*(ll)num);
	}
	return 0;
}

/*
DESCRIPTION:
0 0 0 = 0
1 0 0 = 0
1 1 0 = 1
1 1 1 = 3
2 0 0 = 1
2 1 0 = 3
2 1 1 = 6
2 2 0 = 6
2 2 1 = 10
2 2 2 = 15
*/