这道题开始是想用贪心或者DP求解的,但看它分类在搜索这块,所以就想用DFS。
一开始一直提示TLE,最后搜了一下别人的解题报告,发现思路很相似,不同之处在于他没有考虑重复的路径。
比如说 1 2 3 4 ,当以1为开端时,会把所有有12,13,14的都穷举到,那么到2时,就不用考虑1了,考虑23,24就可以。
代码写的也很有巧妙性,注意在每次DFS时会标记已经使用,所以搜索后要置0。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=1000;
int vis[maxn];
int maxv=0,amount=0,limit;
int n,m;
struct stone
{
int w;
int v;
}s[maxn];
void dfs(int v,int w,int a,int x)
{
if(w>limit||a>amount) return;
if(a==amount) {maxv=max(maxv,v);return;}
for(int i=x;i<n;i++)
{
if(vis[i]==0)
{
vis[i]=1;
dfs(v+s[i].v,w+s[i].w,a+1,i); //也是从i开始,之前的已经确定是1,没有必要再去计算了
vis[i]=0;
}
}
}
int main(void)
{
int i,j,k;
int T;
cin>>T;
while(T--)
{
maxv=0;
memset(vis,0,sizeof(vis));
cin>>n>>amount;
for(i=0;i<n;i++)
cin>>s[i].v>>s[i].w;
cin>>limit;//限制条件,最大重量
for(i=0;i<n;i++)
{
vis[i]=1;
dfs(s[i].v,s[i].w,1,i);
vis[i]=0;
}
cout<<maxv<<endl;
}
}
之后中午睡觉醒来house,想想看完全不用标记,默认从下一块石头处开始计算就可以了,即从i=x+1;i<n;i++处开始。同样AC
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=1000;
int vis[maxn];
int maxv=0,amount=0,limit;
int n,m;
struct stone
{
int w;
int v;
}s[maxn];
void dfs(int v,int w,int a,int x)
{
if(w>limit||a>amount) return;
if(a==amount) {maxv=max(maxv,v);return;}
for(int i=x+1;i<n;i++)
{
if(vis[i]==0)
{
//vis[i]=1;
dfs(v+s[i].v,w+s[i].w,a+1,i); //也是从i开始,之前的已经确定是1,没有必要再去计算了
//vis[i]=0;
}
}
}
int main(void)
{
int i,j,k;
int T;
cin>>T;
while(T--)
{
maxv=0;
memset(vis,0,sizeof(vis));
cin>>n>>amount;
for(i=0;i<n;i++)
cin>>s[i].v>>s[i].w;
cin>>limit;//限制条件,最大重量
for(i=0;i<n;i++)
{
//vis[i]=1;
dfs(s[i].v,s[i].w,1,i);
//vis[i]=0;
}
cout<<maxv<<endl;
}
}