P3905 道路重建

P3905 道路重建
我一开始想错了，我的是类似kruskal，把毁坏的边从小到大加，并且判断联通性。但是这有一个问题，你可能会多加，就是这条边没用，但是它比较小，你也加上了。
居然还有10分，数据也是水水的。。。

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstring>
#define inf 2147483647
#define For(i,a,b) for(register int i=a;i<=b;i++)
#define p(a) putchar(a)
#define g() getchar()
//by war
//2017.10.10
using namespace std;
int n,m,dd,A,B,x,y,v,t;
int a[110][110];
bool b[110][110];
int d[110];
int ans;

struct node
{
int l,r,v;
bool operator<(const node&aa)const
{
return v<aa.v;
}
}e[10000];

void in(int &x)
{
char c=g();x=0;
while(c<'0'||c>'9')c=g();
while(c<='9'&&c>='0')x=x*10+c-'0',c=g();
}
void o(int x)
{
if(x>9)o(x/10);
p(x%10+'0');
}

int find(int x)
{
if(d[x]==x)return x;
d[x]=find(d[x]);
return d[x];
}

int main()
{
in(n),in(m);
For(i,1,n)
d[i]=i;
For(i,1,m)
{
in(x),in(y),in(v);
a[x][y]=v;
a[y][x]=v;
}
in(dd);
For(i,1,dd)
{
in(x),in(y);
b[x][y]=true;
b[y][x]=true;
}
in(A),in(B);
For(i,1,n)
For(j,1,n)
{
if(!b[i][j]&&a[i][j]>0)
d[find(i)]=find(j);
if(b[i][j])
{
e[++t].l=i;
e[t].r=j;
e[t].v=a[i][j];
}
}
sort(e+1,e+t+1);
For(i,1,t)
{
if(find(A)!=find(B))
{
if(find(e[i].l)!=find(e[i].r))
{
d[find(e[i].l)]=find(e[i].r);
ans+=e[i].v;
}
}
else
{
o(ans);
break;
}
}
return 0;
}

正解是把未坏的边的权值设成0，坏的边的值不变，跑spfa即可。

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstring>
#define inf 2147483647
#define len 10010
#define For(i,a,b) for(register int i=a;i<=b;i++)
#define p(a) putchar(a)
#define g() getchar()

using namespace std;
int n,m,x,y,v,dd,A,B;
queue<int>q;
int d[110];
bool vis[110],b[110][110];

struct node
{
int n,v;
node *next;
}*e[len];

void push(int x,int y,int v)
{
node *p;
p=new node();
p->n=y;
p->v=v;
if(e[x]==NULL)
e[x]=p;
else
{
p->next=e[x]->next;
e[x]->next=p;
}
}

void in(int &x)
{
int y=1;
char c=g();x=0;
while(c<'0'||c>'9')
{
if(c=='-')
y=-1;
c=g();
}
while(c<='9'&&c>='0')x=x*10+c-'0',c=g();
x*=y;
}

void o(int x)
{
if(x<0)
{
p('-');
x=-x;
}
if(x>9)o(x/10);
p(x%10+'0');
}

void spfa(int x)
{
For(i,1,n)
d[i]=inf;
d[x]=0;
q.push(x);
node *p;
int t;
while(q.size()>0)
{
t=q.front();
p=e[t];
vis[t]=true;
while(p!=NULL)
{
if(!b[t][p->n])
{
if(d[t]<d[p->n])
{
d[p->n]=d[t];
if(!vis[p->n])
q.push(p->n);
}
}
else
{
if(d[t]+p->v<d[p->n])
{
d[p->n]=d[t]+p->v;
if(!vis[p->n])
q.push(p->n);
}
}
p=p->next;
}
vis[t]=false;
q.pop();
}
}

int main()
{
in(n),in(m);
For(i,1,m)
{
in(x),in(y),in(v);
push(x,y,v);
push(y,x,v);
}
in(dd);
For(i,1,dd)
{
in(x),in(y);
b[x][y]=true;
b[y][x]=true;
}
in(A),in(B);
spfa(A);
o(d[B]),p(' ');
return 0;
}