poj-2531

31 篇文章 0 订阅
// 3916K    375MS   G++
#include <cstdio>  
#include <queue>  
  
using namespace std;  
  
#define MAX 22  
  
int NodeBitId[MAX];  
  
int G[MAX][MAX];  
  
#define INF 99999;  
  
long MAXT;  
  
void initNodeBitId() {  
    int val = 0x1;  
    for (int i = 1; i <= MAX; i++) {  
        // printf("%d\n", val);  
        NodeBitId[i] = val;  
        val <<= 1;  
    }  
}  
  
struct BFSNode {  
    int chooseNum;  
    long T;  
    int choosedMap;  
};  
  
int NodeNum;
#define MAX2 1030*1030

queue<BFSNode> BFSQueue;

char BFSFlag[MAX2];

int BFS() {
    while(BFSQueue.size()) {
        BFSQueue.pop();
    }
    MAXT = -INF;
    BFSNode beginNode;
    beginNode.chooseNum = 0;
    beginNode.T = 0;
    beginNode.choosedMap = 0;
    BFSFlag[0] = 1;
    BFSQueue.push(beginNode);

    int mask = 0xffffffff>>(32 - NodeNum);

    while(BFSQueue.size()) {
        BFSNode curNode = BFSQueue.front();
        BFSQueue.pop();
        long curT = curNode.T;
        // printf("%ld\n", curT);
        MAXT = MAXT > curT ? MAXT : curT;
        int curChooseNum = curNode.chooseNum;
        int curChoosedMap = curNode.choosedMap;

        for (int i = 1; i <= NodeNum; i++) {
            // printf("%d %d\n", NodeBitId[i], curChoosedMap);
           if (! (NodeBitId[i]&curChoosedMap)) { // if node i is not choosed yet
                int newChoosedMap = curChoosedMap|NodeBitId[i];
                int reverseChoosedMap = (~newChoosedMap)&mask;
                if (!BFSFlag[newChoosedMap] && !BFSFlag[reverseChoosedMap]) {
                    BFSFlag[newChoosedMap] = 1;
                    BFSFlag[reverseChoosedMap] = 1;
                    BFSNode newNode;
                    newNode.chooseNum = curChooseNum + 1;
                    newNode.choosedMap = newChoosedMap;
                    long newT = curT;
                    for (int j = 1; j <= NodeNum; j++) {
                       if (! (NodeBitId[j]&curChoosedMap)) { // if node j is not choosed
                            newT += G[j][i];
                       } else { // if j has been choosed
                            newT -= G[j][i];
                       }
                    }
                    newNode.T = newT;
                    BFSQueue.push(newNode);
                }
           } 
        }

    }

}

void solve() {
    BFS();
    printf("%ld", MAXT);
}

int main() {  
    initNodeBitId();  
    while(scanf("%d", &NodeNum) != EOF) {  
        for (int i = 1; i <= NodeNum; i++) {
           for (int j = 1; j <= NodeNum; j++) {
               scanf("%d", &G[i][j]);
           } 
        }
        solve();
    }  
}  


最初看这道题还想着用流或者带权二分图解决,后来想不出怎么解,就老老实实的用BFS解了, 后来看有人用 无向完全图的最大割问题 解决的,不过貌似时间也差不多。

BFS解的步骤,就是每次在已经选择的node中,增加一个新的node,然后看增加以后,traffic T是否更大,如果变大,就更新T,最后遍历完所有的组合以后,输出T就可以。

这道题是纯遍历每个节点(以前的很多BFS是有结束条件的),每个节点对应这一种选取方案,而因为节点数量最多有20个,

因此,用一个int作为bitmap保存此BFS node的选取方案完全足够,不过要事先搞一个数组对应此bitmap中的每一位,即每一个node:

0x1, 0x2, 0x4.... 直到第20位。

BFS的起始node对应的选取方案是一个都不选取到subnetwork1, 然后在此node(选取方案)基础上,可以选取一个新的node,这个新的node可以从已有的node中任意选取,

如果有20个node, 那么就有20种选择,及此BFSNode 可以达到 20个BFSNode(选取一个node的方案)中,

然后重复上面的BFS过程,每次遍历到一个node时, 根据当前node的选取方案看还有哪些node可以选取,将其添加到BFSQueue中,每次新加一个node,根据当前BFSNode的选取方案的得到新的选取方案的traffic T(比如,本次选取了node j,对于已经选取的node i,T -= G[i][j] , 从i到j的流不再算到T中, 对与没有选取的node k, T+= G[j][k], 从j到k的流算到T里了), 同时根据当前方案的chooseBitmap 生成新的方案的chooseBitmap(只需将相应的位 置1 即可),同时看T是否有新的最大值,如果有,更新之。

一个优化就是:选择方案是互逆对等的,比如有3个node, 那么将 node 1选入 subnetwork1 和 将node2,3选入subnetwork1, T是一样的,因此,在BFS遍历时,

也可以将此条件考虑进去,一旦 方案X被遍历过了,那么方案X的逆方案也可以认为遍历过了。这样就大大减少了遍历次数。

就是占的内存有点多,一个BFSFlag就需要 1024*1024的char,其实用bit完全够了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值