hdu2433 spfa+mark[x][u][v]优化
題意:
? ? ? ? ? 刪除每一條邊求最短路的和,每刪除一個就輸出一個和.
??
思路:
? ? ? ? 直接暴力可定TLE了,自己SB的嘗試過,就要剪紙,當每次輸出一個答案的時候我們沒有必要再從新暴力全跑一邊最短路,我們可以開一個數組mark[s][u][v]來標記,當s為起點是邊u,v是否被用過(其實是可能,記錄的時候是更新就假設被用過,但也能達到剪紙的目的),開個sum[i]數組記錄已i為起點時的和,如果被用過那么以s為起點的就得從新跑,否則不用(跑sum[s]也不會變化,所以沒必要跑).這樣能節省很多時間,但自己想想,如果管理員精心設計各種數據,讓你每一條邊都可能被用過那么就TLE了,但估計不會那么變態,卡數據可以,不能變態的卡,不然題目的答案豈不是要唯一化了..
#include<stdio.h>
#include<string.h>
#include<queue>
#define N_node 105
#define N_edge 6100
#define inf ?1000000000
using namespace std;
typedef struct
{
int to ,cost ,next;
}STAR;
STAR E[N_edge];
int list[N_node] ,tot;
int s_x[N_node];
int mark[N_node][N_node][N_node];
int map[N_node][N_node];
int sum[N_node];
int Q[N_edge>>1][2];
void add(int a ,int b ,int c)
{
E[++tot].to = b;
E[tot].cost = c;
E[tot].next = list[a];
list[a] = tot;
}
void spfa(int s ,int n ,int key ,int u ,int v)
{
int mark_q[N_node] = {0};
mark_q[s] = 1;
for(int i = 0 ;i <= n ;i ++)
s_x[i] = inf;
s_x[s] = 0;
queue<int>q;
q.push(s);
while(!q.empty())
{
int xin ,tou;
tou = q.front();
q.pop();
mark_q[tou] = 0;
for(int k = list[tou] ;k ;k = E[k].next)
{
xin = E[k].to;
if(u == tou && v == xin || u == xin && v == tou)
continue;
if(s_x[xin] > s_x[tou] + E[k].cost)
{
s_x[xin] = s_x[tou] + E[k].cost;
if(key == 1)
mark[s][tou][xin] = mark[s][xin][tou] = 1;
if(!mark_q[xin])
{
mark_q[xin] = 1;
q.push(xin);
}
}
}
}
return ;
}
int main ()
{
int n ,m ,i ,a ,b ,ii ,jj;
while(~scanf("%d %d" ,&n ,&m))
{
memset(list ,0 ,sizeof(list));
memset(map ,0 ,sizeof(map));
tot = 1;
for(i = 1 ;i <= m ;i ++)
{
scanf("%d %d" ,&a ,&b);
map[a][b] ++;
map[b][a] ++;
add(a ,b ,1);
add(b ,a ,1);
Q[i][0] = a;
Q[i][1] = b;
}
int first = 0;
memset(sum ,0 ,sizeof(sum));
memset(mark ,0 ,sizeof(mark));
for(i = 1 ;i <= n ;i ++)
{
spfa(i ,n ,1 ,0 ,0);
for(ii = 1 ;ii <= n ;ii ++)
{
sum[i] += s_x[ii];
if(s_x[ii] == inf)
break;
}
first += sum[i];
if(ii != n + 1)
break;
}
if(i != n + 1)
{
for(i = 1 ;i <= m ;i ++)
printf("INF\n");
continue;
}
for(i = 1 ;i <= m ;i ++)
{
a = Q[i][0];
b = Q[i][1];
if(map[a][b] > 1)
{
printf("%d\n" ,first);
continue;
}
int now = first;
int kk = 0;
for(ii = 1 ;ii <= n ;ii ++)
{
if(mark[ii][a][b])
{
spfa(ii ,n ,0 ,a ,b);
int s = 0;
for(jj = 1 ;jj <= n ;jj ++)
{
s += s_x[jj];
if(s_x[jj] == inf)
break;
}
now = now - sum[ii] + s;
if(jj != n + 1)
break;
}
}
if(ii != n + 1)
{
printf("INF\n");
continue;
}
printf("%d\n" ,now);
}
}
return 0;
}
總結
以上是生活随笔為你收集整理的hdu2433 spfa+mark[x][u][v]优化的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: hdu3986 spfa + 枚举最短路
- 下一篇: hdu1572 水搜索