P2196 [NOIP1996 提高组] 挖地雷 ——DFS或DP

题目链接: 点我(传送门)

分析: 数据范围这么小,深搜肯定能过,一发搜索

解法一:DFS

AC代码

#include<bits/stdc++.h>
#define QUICK ios::sync_with_stdio(false),cin.tie(0)
#define MAXN 30
using namespace std;
int e[MAXN][MAXN],mine[MAXN];
int vis[MAXN],path[MAXN],ans[MAXN];
int n,maxm,len;//max_mine

int check(int x)
{
    
    
    for(int i=1;i<=n;i++)
        if(e[x][i]&&!vis[i])return 0;
    return 1;
}

void dfs(int x,int step,int num)
{
    
    
    if(check(x)){
    
    
        if(num>maxm){
    
    
            maxm=num;len=step;
            for(int i=1;i<=step;i++)ans[i]=path[i];
        }
    }
    for(int i=1;i<=n;i++){
    
    
        if(e[x][i]){
    
    
            vis[i]=1;
            path[step+1]=i;
            dfs(i,step+1,num+mine[i]);
            vis[i]=0;//recall
        }
    }

}

int main()
{
    
    
    QUICK;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>mine[i];
    for(int i=1;i<=n-1;i++)for(int j=i+1;j<=n;j++)cin>>e[i][j];
    for(int i=1;i<=n;i++){
    
    
        memset(vis,0,sizeof(vis));
        vis[i]=1;path[1]=i;
        dfs(i,1,mine[i]);
    }
    for(int i=1;i<=len;i++)cout<<ans[i]<<" ";
    return  cout<<"\n"<<maxm<<endl,0;
}

解法二:DP

dp[i] 的含义是:当路径以序号为 i 的节点做终点时的最大地雷数。

AC代码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 30
int dp[MAXN],p[MAXN],e[MAXN][MAXN],mine[MAXN];
int n,max_mine,maxi;
void pr(int x)
{
    
    
    if(x==0)return ;
    pr(p[x]);
    cout<<x<<" ";
}

int main()
{
    
    
    cin>>n;
    for(int i=1;i<=n;i++){
    
    cin>>mine[i];dp[i]=mine[i];}
    for(int i=1;i<=n-1;i++)for(int j=i+1;j<=n;j++)cin>>e[i][j];
    for(int i=1;i<=n;i++){
    
    
        for(int j=1;j<i;j++)
            if(e[j][i])if(dp[i]<dp[j]+mine[i])dp[i]=dp[j]+mine[i],p[i]=j;
        if(max_mine<dp[i])maxi=i,max_mine=dp[i];
    }
    pr(maxi);cout<<endl;
    return cout<<max_mine<<endl,0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43615816/article/details/114928939