一个疑问
查看原帖
一个疑问
1121518
StarsIntoSea楼主2025/7/28 19:55

下面是一份 AC 的代码,如果把第 26 行的注释取掉就会 WA。自己手搓小样例都过不了。

加上这一行应该对正确性没有影响啊?

小样例在代码里。

#include <string.h>
#include <iostream>
using namespace std;
#define int long long
const int N=205;
int n=0,tmp,t,s,e,inf=1e16;
int num[1005];
struct marx{
	int d[N][N];
};
marx mul(marx a,marx b){
	marx s;
	for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) s.d[i][j]=inf;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			for(int k=1;k<=n;++k)	
				s.d[i][j]=min(s.d[i][j],a.d[i][k]+b.d[k][j]);
	return s;
}

signed main(){
	scanf("%lld%lld%lld%lld",&tmp,&t,&s,&e);
	marx a,b;
	memset(a.d,0x3f,sizeof a.d);
//	for(int i=1;i<=200;++i) for(int j=1;j<=200;++j) a.d[i][j]=b.d[i][j]=inf;
//	for(int i=1;i<=200;++i) a.d[i][i]=b.d[i][i]=0;
	for(int i=1;i<=t;++i){
	    int w,u,v; scanf("%lld%lld%lld",&w,&u,&v);
	    if(!num[u]) num[u]=++n; //离散化
	    if(!num[v]) num[v]=++n;
		a.d[num[u]][num[v]]=a.d[num[v]][num[u]]=w;
	}
	--tmp; b=a;
	while(tmp){
		if(tmp&(int)1) a=mul(a,b);
		b=mul(b,b);
		tmp>>=1;
	}
	printf("%lld\n",a.d[num[s]][num[e]]);
}

/*
  小样例输入:
  6 7 1 5
  15 1 5
  6 1 2
  7 2 5
  1 1 3
  3 3 2
  4 4 2
  2 4 5
  
  正确输出:
  12
 */
2025/7/28 19:55
加载中...