1. 程式人生 > 實用技巧 >Acwing 165 小貓爬山 (dfs)

Acwing 165 小貓爬山 (dfs)

題面

翰翰和達達飼養了N只小貓,這天,小貓們要去爬山。

經歷了千辛萬苦,小貓們終於爬上了山頂,但是疲倦的它們再也不想徒步走下山了(嗚咕>_<)。

翰翰和達達只好花錢讓它們坐索道下山。

索道上的纜車最大承重量為W,而N只小貓的重量分別是C1、C2……CN。

當然,每輛纜車上的小貓的重量之和不能超過W。

每租用一輛纜車,翰翰和達達就要付1美元,所以他們想知道,最少需要付多少美元才能把這N只小貓都運送下山?

輸入格式
第1行:包含兩個用空格隔開的整數,N和W。

第2..N+1行:每行一個整數,其中第i+1行的整數表示第i只小貓的重量Ci。

輸出格式
輸出一個整數,表示最少需要多少美元,也就是最少需要多少輛纜車。

資料範圍
1≤N≤18,
1≤Ci≤W≤108
輸入樣例:
5 1996
1
2
1994
12
29
輸出樣例:
2

思路

搜尋,好吧,我連搜尋策略都找不到對。往往剛開始的想法是一輛一輛搜,不停裝貓,但是可惜的是這個策略並不正確,因為可能我這個車有餘量,而且這個餘量可以裝下後面的小貓,但這種情況會被我們忽略。所以我們換一種,用一個數組儲存每個車的量,每次都去搜一遍所有車,有餘量就裝,沒有就新開一個。

程式碼實現

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<set>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=20;
ll a[maxn];
ll p[maxn];
ll n,mostval;
long long ans=1e9;
bool cmp (ll a,ll b)  {
    return a>b;
}
inline void dfs (long long x,long long cnt) {
    if (cnt>=ans)  return ;
    if (x==n+1) {
        ans=min (cnt,ans);
        return ;
    }
    for (int i=1;i<=cnt;i++) {
        if (p[i]+a[x]<=mostval) {
            p[i]+=a[x];
            dfs (x+1,cnt);
            p[i]-=a[x];
        }
    }
    p[cnt+1]=a[x];
    dfs (x+1,cnt+1);
    p[cnt+1]=0;
}
int main () {
    cin>>n>>mostval;
    for (int i=1;i<=n;i++) cin>>a[i];
    sort (a+1,a+1+n,cmp);
    dfs (1,0);
    cout<<ans<<endl;
    return 0;
}