砝碼稱重2
阿新 • • 發佈:2019-01-23
題目描述 Description
有n個砝碼,現在要稱一個質量為m的物體,請問最少需要挑出幾個砝碼來稱?
注意一個砝碼最多隻能挑一次
輸入描述 Input Description
第一行兩個整數n和m,接下來n行每行一個整數表示每個砝碼的重量。
輸出描述 Output Description
輸出選擇的砝碼的總數k,你的程式必須使得k儘量的小。
樣例輸入 Sample Input
3 10
5
9
1
樣例輸出 Sample Output
2
資料範圍及提示 Data Size & Hint
1<=n<=30,1<=m<=2^31,1<=每個砝碼的質量<=2^30
時間限制:1S 空間限制:16MB
迭代,第一種方法,維護字尾陣列剪紙,dfs暴力。
#include<iostream>
#include<algorithm>
using namespace std;
long long s=0,o=0;long long n,m;
long long a[50],f[50];
bool cmp(int x,int y)
{ return x>y;
}
void dfs(int t,int now,int ans)
{ if (s==m)
{ o=1;
return;
}
if (t>ans)
return;
if (s+f[now]-f[now+ans-t+1]<m)
return;
if (s+f[n-ans+t]>m)
return;
for (int i=now;i<=n-ans+t;++i)
{ s+=a[i];
dfs(t+1,i+1,ans);
s-=a[i];
}
}
int main()
{
cin>>n>>m;
for (int i=1;i<=n;++i)
cin>>a[i];
sort(a+1,a+1+n,cmp);
f[n]=a[n];
for (int i=n-1;i>=1;--i)
{ f[i]=a[i];
f[i]+=f[i+1];
}
for (int i=1;i<=n;++i)
{ s=0;
dfs(1,1,i);
if (o)
{ cout<<i;
return 0;
}
}
}
還有自己用map實現的code
#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
long long a[50];
long long n,m,r;int ans=10000;
map<long,int> M;
bool cmp(int x,int y)
{ return x>y;
}
void dfs(int t,int now,int sum)
{ M[sum]=t;
for (int i=now;i<=r;++i)
dfs(t+1,i+1,sum+a[i]);
}
void dfs2(int t,int now,int sum)
{ if (M.find(m-sum)!=M.end())
ans=min(ans,t+M[m-sum]);
for (int i=now;i<=n;++i)
dfs2(t+1,i+1,sum+a[i]);
}
int main()
{ cin>>n>>m;
for (int i=1;i<=n;++i)
cin>>a[i];
sort(a+1,a+1+n,cmp);
r=n/2;
dfs(0,1,0);
dfs2(0,n/2+1,0);
cout<<ans;
}