// 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完全够了。