oj #1842 斗地主

题目描述

alvin自从开了游乐园之后，经营的风生水起，没过多久就收回了成本.为了创建一个高端的娱乐帝国，alvin打算再开一个赌场，并且里面只会有alvin认为最好玩的游戏–斗地主. 开张第一天便生意火爆，有n个人来赌场玩斗地主.然而在斗地主界有一个规矩，那就是同一桌的三个人带的筹码的异或值为0，才能使参与游戏的三个人都感到愉快，并且称这一桌为“nice round”. 比如，若一共有三个人，所携带的筹码数分别为1,2,3，则 = 0，这便是一个“nice round”.若分别携带的筹码为1，2，4，则不是. 现在alvin的有个问题，这n个顾客可以组成多少个不同的“nice round”呢？ Input

输入描述

有多组数据输入. 每组数据第一行包含一个整数n，代表有n个顾客. 接下来一行每行包含n个非负整数，代表n个顾客所携带的筹码数，其中ai代表第i个顾客的筹码数. 数据规模： 对于100%的数据，保证ai <= 500; 对于40%的数据，保证n <= 200; 对于100的数据，保证n <= 5000.

输出描述

对于每组数据，输出一个整数，代表能够产生的不同的nice round的组数.每组一行.

样例输入

3
1 2 3
4
1 1 2 3

样例输出

1
2

题目解释：

1.因为ai的值小于500，所以开一个505的数组，初始化为0

2.循环读取输入的n的值，直到没有更多输入

3.计算形如(0, i, i)的三元组数量，并累加到num中，例：ai出现的次数是x,从x中任选2个。Cx2

4.计算形如(0, 0, 0)的三元组数量，并累加到num中，例：0出现的次数是x,从x中任选3个。Cx3

5.计算形如(i, j, k)的三元组数量，并累加到num中，

6.最后输出num,并将数组和num初始化

代码：

#include <bits/stdc++.h> 
using namespace std;

int n,k;

int main() {
    vector<long long> a(505, 0); // 初始化时直接全部设为 0    
  
    while (cin >> n) {  
        // 使用 fill 函数将 a的所有元素重置为0  
        fill(a.begin(), a.end(), 0); 

    	long long int num=0;
    	for(int i=0;i<n;i++){
    		cin>>k;
    		a[k]++;
		}

		if(a[0]){
			for(int i=1;i<500;i++){
				if(a[i]>1) 
                    num+=a[i]*(a[i]-1)/2*a[0];
			}
		}

		if(a[0]>2) 
            num+=a[0]*(a[0]-1)*(a[0]-2)/6;
		
		for(int i=0;i<500;i++){
			for(int j=i+1;j<500;j++){
				for(int k=j+1;k<500;k++){
					 if ((i ^ j ^ k) == 0) { 
                        num += a[i] * a[j] * a[k];
                    }
				}
			}
		}
		cout<<num<<endl;
	}
    return 0;
}