[华为oj]称砝码

算法本身效率很低，但感觉比较简洁，所以自己记录了下。华为的测试用例才一个，通过是通过了，但觉得根本没法检查这段程序的正确性，个人觉得逻辑上没错。

#include <iostream>
#include <set>
#include <algorithm>

using namespace std;

int fama(int n,int* weight,int* nums)
{
    set<int> setInt;
    setInt.insert(0);
    for(int i=0;i<n;i++)
    {
        int w=weight[i];
        set<int> setTemp;
        set<int>::iterator bitr=setInt.begin();
        set<int>::iterator eitr=setInt.end();
        for(set<int>::iterator p=bitr;p!=eitr;p++)
        {
            int wt=*p;
            for(int  count=0;count<=nums[i];count++)
            {
                int twt=wt+w*count;
                setTemp.insert(twt);
            }
        }
        setInt.insert(setTemp.begin(),setTemp.end());
    }
    return static_cast<int>(setInt.size());
}

int main()
{
    int* weight;
    int* nums;
    int n;
    cin>>n;

    weight=new int[n];
    nums=new int[n];

    for(int i=0;i<n;i++)
        cin>>weight[i];

    for(int i=0;i<n;i++)
        cin>>nums[i];

    cout<<fama(n,weight,nums)<<endl;

}