问题描述:输入整数数组arr,找出其中最小的k个数。
思路:排序后求解:
public int[]lastK(int []nums,int k)
{
Arrays.sort(nums);
int []ret=new int[k];
for(int i=0;i<k;i++)
{
ret[i]=nums[i];
}
return ret;
}
//使用最大堆求解
public int []getLast(int []nums,int k)
{
PriorityQueue<Integer>maxHeap=new PriorityQueue<>((a,b)->{return b-a;});
for(int i=0;i<k;i++)
{
maxHeap.offer(nums[i]);
}
for(int i=k;i<nums.length;i++)
{
if(maxHeap.peek()>nums[i])
{
maxHeap.poll();maxHeap.offer(nums[i]);
}
}
int []res=new int[k];
for(int i=0;i<k;i++)
{
res[i]=queue.poll();
}
return ret;
}
问题描述:给你一个m*n的网格accounts,其中accounts[i][j]是第i位顾客在第j家银行托管的资产数量,返回最富有的客户所有的资产总量。
public int maxWealth(int [][]nums)
{
int mostWealth=0;
for(int i=0;i<nums.length;i++)
{
int current=0;
for(int j=0;j<nums[i].length;j++)
{
current+=nums[j];
}
mostWealth=Math.max(mostWealth,current);
}
return mostWealth;
}