贪心——IQ

题目描述

根据世界某权威学会的一项调查，学信息学的学生IQ非常高。举个最好的例子，如果我们把学信息学的一些学生调去学数学，那么两个竞赛的学生平均IQ都会提升！！
现在给出一群数学竞赛全体学生的IQ和信息学竞赛全体学生的IQ，问最多能把几个学信息学的学生调去学数学，而两个竞赛的学生平均IQ都有提升呢？
每次只调一个学生，而且每次调配，两个竞赛的学生平均IQ都要提升。

输入

第1行一个正整数N(N≤I00)，代表数学竞赛班学生的数目；
第2行，有N个正整数，代表每个数学学生的IQ（1到200之间）；
第3行，一个整数M(M≤100)，代表信息学竞赛班学生数目；
第4行，有M个正整数，代表每个信息学竞赛班学生的IQ（1到200之间）。
输入数据保证信息学学生平均IQ高于数学学生。

输出

一个正整数，代表能从信息学调去学数学的学生的个数。

样例输入

复制样例数据

3
3 2 3
3
3 2 5

样例输出

ps:挺简单的，代码要清楚明了啊，不然也许一些低级错误会出现。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[1000],b[1000];
int main()
{
    int n,m,ans=0;
    cin>>n;
    int i,suma=0;
    for(i=1;i<=n;i++){
        cin>>a[i];
        suma+=a[i];
    }
    double ave1=suma*1.0/n;
    cin>>m;
    int sumb=0;
    for(i=1;i<=m;i++){
        cin>>b[i];
        sumb+=b[i];
    }
    double ave2=sumb*1.0/m;
    sort(b+1,b+m+1);
    /*
    bool t=true;
    while(t){
        t=false;
        for(i=1;i<=m;i++){
            if(b[i]<ave2&&b[i]>ave1&&b[i]!=201){
                t=true;
                suma+=b[i];
                sumb-=b[i];
                b[i]=201;
                break;
            }
        }
        if(t) ans++;
        ave1=suma*1.0/((n+ans)*1.0);
        ave2=sumb*1.0/((m-ans)*1.0);
    }
    */
    int m1=m;
    for(i=1;i<=m1;i++){
        if(b[i]>ave1&&b[i]<ave2){
            ans++;
            m--;
            n++;
            suma+=b[i];
            sumb-=b[i];
            ave1=suma*1.0/(n*1.0);
            ave2=sumb*1.0/(m*1.0);
        }
    }
    cout<<ans<<endl;
    return 0;
}