阿里巴巴在线测试题
前幾天做了一個阿里巴巴關于小猴子摘桃子的問題,現在我整理一下我的思路,希望各位大神給予指教。
1、根據每棵樹的桃子的數量以及該樹的位置,計算該樹的權重,計算公式如下:
w=v*i
其中:v為每棵樹的價值,i為每棵樹的位置。
v是根據每棵樹桃子的數量排序計算出來的。
2、權重計算出來后,將權重進行排序;
3、首先從權重最大的進行摘取,然后接著比較剩余的樹,如果該樹的價值大于當前的價值,并且該樹的位置大于當前的位置,則摘取,否則不摘取。
通過以上思路的整理,代碼如下:
public class Monkey {
public static void main(String[] args) {
int[] datas={10,4,5,12,8};
int[] flags=new int[5];
int[][] wights=comoute(datas);//計算權重
for(int i=0;i<wights.length;i++){
System.out.print("權重:"+wights[i][0]+", 價值: "+wights[i][1]+", 位置: "+wights[i][2]);
System.out.print("\n");
}
selPeach(wights,flags);
int count=0;
System.out.println("摘取的位置: ");
for(int i=0;i<flags.length;i++){
if(flags[i]==1){
count++;
System.out.println(i);
}
}
System.out.println("摘取總量:"+count);
}
/**
* 計算權重
* @param datas
* @return
*/
private static int[][] comoute(int[] datas){
int[] datasTemp=new int[datas.length];
for(int i=0;i<datas.length;i++){
datasTemp[i]=datas[i];
}
sort(datas,0,datas.length-1);
int[][] wights=new int[datas.length][3];
for(int i=0;i<datasTemp.length;i++){
int index=look(datas, datasTemp[i]);
wights[i][0]=(index+1)*(datasTemp.length-i);//通過樹的位置和價值計算權重
wights[i][1]=datasTemp[i];
wights[i][2]=i;
}
return wights;
}
/**
* 快速排序
* @param datas
* @param low
* @param high
* @return
*/
private static int[] sort(int[] datas, int low, int high){
int lo=low;
int hi=high;
if(lo<hi){
int key=datas[lo];
while(lo<hi){
while(datas[hi]<=key && lo<hi){
hi--;
}
datas[lo]=datas[hi];
while(datas[lo]>=key && lo<hi){
lo++;
}
datas[hi]=datas[lo];
}
datas[hi]=key;
sort(datas,low,hi-1);
sort(datas,hi+1,high);
}
return datas;
}
/**
* 折半查找
* @param datas
* @param key
* @return
*/
private static int look(int[] datas, int key){
int low=0;
int high=datas.length-1;
int mid=0;
while(low<=high){
mid=(low+high)/2;
if(datas[mid]>key){
low=mid+1;
}else if(datas[mid]<key){
high=mid-1;
}else{
break;
}
}
return mid;
}
/**
* 選擇樹
* @param wights
* @param flags
* @return
*/
private static int[] selPeach(int[][] wights, int[] flags){
sortrows(wights,0,wights.length-1,0);//將權重進行排序
System.out.println("################################");
for(int i=0;i<wights.length;i++){
System.out.print("權重:"+wights[i][0]+", 價值: "+wights[i][1]+", 位置: "+wights[i][2]);
System.out.print("\n");
}
int currentValue=wights[0][1];
int currentIndex=wights[0][2];
for(int i=0;i<wights.length;i++){
if(wights[i][1]>=currentValue && wights[i][2]>=currentIndex){
flags[wights[i][2]]=1;
currentValue=wights[i][1];
currentIndex=wights[i][2];
}
}
return flags;
}
/**
* 快速排序,并更新其他列
* @param wights
* @param low
* @param high
* @param col
* @return
*/
private static int[][] sortrows(int[][] wights, int low, int high, int col){
int lo=low;
int hi=high;
if(lo<hi){
int[] keys=new int[wights[0].length];
for(int i=0;i<wights[0].length;i++){
keys[i]=wights[lo][i];
}
while(lo<hi){
while(wights[hi][col]<=keys[col] && lo<hi){
hi--;
}
for(int i=0;i<wights[0].length;i++){
wights[lo][i]=wights[hi][i];
}
while(wights[lo][col]>=keys[col] && lo<hi){
lo++;
}
for(int i=0;i<wights[0].length;i++){
wights[hi][i]=wights[lo][i];
}
}
for(int i=0;i<wights[0].length;i++){
wights[hi][i]=keys[i];
}
sortrows(wights, low, hi-1, col);
sortrows(wights, hi+1, high, col);
}
return wights;
}
}
總結
- 上一篇: 会让你变得与众不同的22个技巧
- 下一篇: C++加载csb文件