首页 > 代码库 > PKU 2531 Network Saboteur(dfs+剪枝||随机化算法)

PKU 2531 Network Saboteur(dfs+剪枝||随机化算法)

题目大意:原题链接

给定n个节点,任意两个节点之间有权值,把这n个节点分成A,B两个集合,使得A集合中的每一节点与B集合中的每一节点两两结合(即有|A|*|B|种结合方式)权值之和最大。

标记:A集合:true  B集合:false 

解法一:dfs+剪枝

#include<iostream>
#include<cstring>
using namespace std;
int n,ans;
bool in[25];
int graph[25][25];
void dfs(int i,int cursum)
{
    in[i]=true;
    int maxsum=cursum;
    for(int j=1;j<=n;j++){
        if(!in[j]) maxsum+=graph[i][j];
        else maxsum-=graph[i][j];
    }
    if(maxsum>ans) ans=maxsum;
    if(maxsum>cursum){//枚举+递归 
        for(int j=i+1;j<=n;j++)
            dfs(j,maxsum);
    }//如果加进A组没有使得maxsum增大,则不加进去
    in[i]=false; 
}
int main()
{
    cin>>n;
    memset(in,0,sizeof(in));
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            cin>>graph[i][j];
    }
    ans=0;
    dfs(1,0);
    cout<<ans<<endl;
}

解法二:随机化算法(好神奇的思路)

#include<iostream>
#include<cstdlib>  
using namespace std;  
const int TLE=2000;//本题时间限制为2000ms   
int main()  
{  
    int n,m[25][25]; 
    cin>>n; 
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            cin>>m[i][j];   
    }
    /*Random Algorithm*/  
    bool in[25]={false};     
    int time=100*TLE;//使随机次数尽可能大,随机结果尽可能接近最优解  
    int maxsum=0,cursum=0;  
    while(time--){  
        int x=rand()%n+1;//范围:1~n    
        in[x]=!in[x];//改变x所在的集合位置        
        for(int i=1;i<=n;i++){
            if(in[i]!=in[x]) cursum+=m[i][x];          
            else  cursum-=m[i][x];
        }
        if(maxsum<cursum) maxsum=cursum;     
    }
    cout<<maxsum<<endl;
}

PKU 2531 Network Saboteur(dfs+剪枝||随机化算法)