[POI] 10th stage I - pB Chocolate

#include<stdio.h>
#include<algorithm>
using namespace std;
int A[1005],B[1005];
int main(){
    int i,j,k,n,m,x=1,y=1,ans=0;
    scanf("%d%d",&n,&m);
    n--,m--;
    for(i=0;i<n;i++)
        scanf("%d",&A[i]);
    for(i=0;i<m;i++)
        scanf("%d",&B[i]);
    sort(A,A+n);
    sort(B,B+m);
    k=n+m,n--,m--;
    while(k--){
        if(n>=0 && A[n]>B[m]){
            ans+=A[n]*y;n--;
            x++;
        }
        else if(m>=0 && A[n]<B[m]){
            ans+=B[m]*x;m--;
            y++;
        }
        else{
            if(n>=0 && x>y){
                ans+=A[n]*y;n--;
                x++;
            }
            else if(m>=0 && x<=y){
                ans+=B[m]*x;m--;
                y++;
            }
        }
    }
    printf("%d\n",ans);
}