XTU-OJ 1386-彩球

彩球

题目描述

有n个球，标号为1∼n，第i个球有的颜色为ci。你需要选择3个不同颜色的球，问有多少种不同的选择方案？

输入格式

第一行是一个整数T (1≤T≤100)，表示输入样例的个数。 每个样例的第一行是一个整数n (1≤n≤104)，表示球的数量。第二行是n个整数ci (1≤ci≤224)，表示球的颜色。

输出格式

每行输出一个样例的结果，为一个整数。

样例输入

2
5
1 2 3 2 1
5
1 2 1 2 1

样例输出

4
0

解题思路：本题有两种写法：

一种是采用递推的方式：3种颜色的球的方案数 = 已经找到的2种颜色的球*当前颜色球的个数，要 求总共有多少种方案数， 就是用累加： +=  ，2种颜色的方案数也一样的操作。  但是注意： tag[3] 、tag[2] 、tag[1] 等式的顺序不能变。 必须是先 tag[3] += 前面的2种颜色方案数*当前颜色数。 计算完之后再更新 tag[2] 的值。

第二种方法是 简单的排列组合问题。 开始直接 C(n,3)，找到总的排列方案数。然后这里面肯定有颜色重复的，所以剩下的就是去重。for遍历  (int i = 0; i < count; i ++) ，判定每一种颜色个数,减去 C(ni,3)+C(ni,2)*(count-2) 。 

AC代码：（递推）

#include <stdio.h>
#include <stdlib.h>


int cmp(const void* p1,const void* p2){
    return *(int *)p1 - *(int *)p2;
}

int main()
{
    int T,n,cnt,count;
    scanf("%d",&T);
    while ( T --)
    {
        __int64 tag[5] = {0};
        int balls[10010] = {0}, color[10010] = {0};
        cnt = 1, count = 0;

        scanf("%d",&n);                       // 输入
        for (int i = 0; i < n; i ++)
            scanf("%d",&balls[i]);

        qsort(balls,n,sizeof(balls[0]),cmp);  // 排序
        for (int i = 1; i <= n; i ++)         // 统计颜色数量
        {
            if  (balls[i] == balls[i-1])  cnt ++;
            else {color[count++] = cnt; cnt = 1;}
        }

        for (int i = 0; i < count; i ++)      // 递推
        {
            tag[3] += tag[2]*color[i];        // 三种颜色球的方案数 += 两种颜色球的方案数*当前颜色球的个数
            tag[2] += tag[1]*color[i];        // 两种颜色球的方案数 += 一种颜色球的方案数*当前颜色球的个数
            tag[1] += color[i];               // 一种颜色球的方案数 += 当前颜色球的个数
        } 
        printf("%I64d\n",tag[3]);
    }
    return 0;
}