小貓爬山(dfs)
阿新 • • 發佈:2018-12-16
小貓爬山
時間限制: 1 Sec 記憶體限制: 128 MB
題目描述
Freda和rainbow飼養了N只小貓,這天,小貓們要去爬山。經歷了千辛萬苦,小貓們終於爬上了山頂,但是疲倦的它們再也不想徒步走下山了(嗚咕>_<)。
Freda和rainbow只好花錢讓它們坐索道下山。索道上的纜車最大承重量為W,而N只小貓的重量分別是C1、C2……CN。當然,每輛纜車上的小貓的重量之和不能超過W。每租用一輛纜車,Freda和rainbow就要付1美元,所以他們想知道,最少需要付多少美元才能把這N只小貓都運送下山?
輸入
第一行包含兩個用空格隔開的整數,N和W。
接下來N行每行一個整數,其中第i+1行的整數表示第i只小貓的重量Ci。
輸出
輸出一個整數,最少需要多少美元,也就是最少需要多少輛纜車。
樣例輸入
複製樣例資料
5 1996 1 2 1994 12 29
樣例輸出
2
提示
對於100%的資料,1<=N<=18,1<=Ci<=W<=10^8。
最近也不知道怎麼了,不在狀態,這種dfs都弄了很久,還是看別人的,下面附上我超時的程式碼,難受,我是列舉每個重量,其實只要列舉次數就行了。
/**/ #include <cstdio> #include <cstring> #include <cmath> #include <cctype> #include <iostream> #include <algorithm> #include <map> #include <set> #include <vector> #include <string> #include <stack> #include <queue> typedef long long LL; using namespace std; int n, w, a[20]; int vis[20], ans = 20; int Sum = 0; bool cmp(int x, int y){ return x > y; } void dfs(int num, int sum, int cnt, int tot){ if(ans <= cnt) return ; if(num == n){ ans = min(ans, cnt); return ; } int f = 0; for (int i = 1; i <= n; i++){ if(vis[i]) continue; vis[i] = 1; if(sum + a[i] > w){ vis[i] = 0; continue; } f = 1; dfs(num + 1, sum + a[i], cnt, tot + a[i]); vis[i] = 0; } if(!f && cnt + ceil(1.0 * (Sum - tot) / w) < ans) dfs(num, 0, cnt + 1, tot); } int main() { //freopen("in.txt", "r", stdin); //freopen("out.txt", "w", stdout); scanf("%d %d", &n, &w); for (int i = 1; i <= n; i++){ scanf("%d", &a[i]); Sum += a[i]; } sort(a + 1, a + n + 1, cmp); dfs(0, 0, 0, 0); printf("%d\n", ans + 1); return 0; } /**/
這是正確的程式碼
/**/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <stack>
#include <queue>
typedef long long LL;
using namespace std;
int n, w;
int a[20], p[20], sum, f;
void dfs(int n, int num){
if(f) return;
if(num == n){
f = 1;
return ;
}
for (int i = 1; i <= n; i++){
if(p[i] + a[i] <= w){
p[i] += a[i];
dfs(n, num + 1);
p[i] -= a[i];
if(f) return ;
}
}
}
int main()
{
//freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
scanf("%d %d", &n, &w);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]), sum += a[i];
sort(a + 1, a + 1 + n, greater<int>());
for (int i = ceil(1.0 * sum / w); i <= n; i++){
f = 0;
dfs(i, 0);
if(f){
printf("%d\n", i);
break;
}
}
return 0;
}
/**/