Help me in solving ZCO14003 problem

My issue

My code

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

int main() {
    long long int n;
	cin>>n;
	long long int arr[1000000];
	for(int i=0;i<n;i++)
	{
	    cin>>arr[i];
	}
// 	int res[n];
// 	for(int i=0;i<n;i++)
// 	{
// 	    int count=0;
// 	    for(int j=0;j<n;j++)
// 	    {
// 	      if(arr[j]>=arr[i])
// 	      {
// 	          count=count+1;
// 	      }
// 	      res[i]=arr[i]*count;
// 	    }
// 	}
sort(arr,arr+n);
int count=0;
int ans=0;
for(int i=0;i<n;i++)
{
   count=arr[i]*(n-i);
   ans=max(ans,count);
    
}
cout<<ans;	
// 	for(int i=0;i<n;i++)
// 	{
// 	    cout<<res[i]<<endl;
// 	}
// 	sort(res,res+n);
// 	cout<<res[n-1];


	
	return 0;
}

Problem Link: CodeChef: Practical coding for everyone

@neelakshi12
Use long long int for ans and count rest it perfectly fine.
include
include
#include<bits/stdc++.h>
using namespace std;

int main() {
long long int n;
cin>>n;
long long int arr[1000000];
for(int i=0;i<n;i++)
{
cin>>arr[i];
}
// int res[n];
// for(int i=0;i<n;i++)
// {
// int count=0;
// for(int j=0;j<n;j++)
// {
// if(arr[j]>=arr[i])
// {
// count=count+1;
// }
// res[i]=arr[i]count;
// }
// }
sort(arr,arr+n);
long long int count=0;
long long int ans=0;
for(int i=0;i<n;i++)
{
count=arr[i]
(n-i);
ans=max(ans,count);

}
cout<<ans;
// for(int i=0;i<n;i++)
// {
// cout<<res[i]<<endl;
// }
// sort(res,res+n);
// cout<<res[n-1];

return 0;

}