SRM 370 DIV 2 (500)

SRM 370 DIV 2 (500)
문제 요약 : 색깔이 중복될 것으로 예상되는 마블의 수를 구하라.

#include <iostream>

#include <vector>

using namespace std;

 

class DrawingMarbles

{

public:

　　double sameColor(vector<int> colors, int n);

};

 

double DrawingMarbles::sameColor(vector<int> colors, int n)

{

　　int i = 0, j = 0;

　　int total = 0;

　　double res = 0;

　　double sum = 1;

 

　　if(colors.size() != 1)

　　{

　　　　　for(i = 0 ; i < colors.size() ; i++)

　　　　　　　　　total += colors[i];

　　　　　for(i = 0 ; i < colors.size() ; i++)

　　　　　{

　　　　　　　　for(j = 0 ; j < n ; j++)

　　　　　　　　　　sum *= ((double)colors[i] - (double)j)
　　　　　　　　　　　　　　　　　　　　　        / ((double)total - (double)j);

　　　　　　　　　

　　　　　　　　res += sum;

　　　　　　　　sum = 1;

　　　　　}

　　}

　　else res = 1.0;

　　return res;

}