2021-08-08

传送门

题目大意

题解

dijkstra算法
先用dijkstra跑一遍正确答案，再用错误代码跑一遍，对比得出答案。但错误算法的复杂度为$O(n^3)$，所以需要优化
1：只找需要的处理
2：当正确答案是无穷大时不处理

优化后复杂度为$O(n^2+nm)$

code

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int w[2010][2010];
vector<pair<int,int>> g[2010];
void dij(int s,int *dis)
{
    priority_queue< pair<int,int>,vector<pair<int,int> >, greater<pair<int,int>> >q;
    q.push(make_pair(0,s));
    dis[s]=0;
    int x,y,k;
    while(!q.empty())
	{
        tie(k,x)=q.top(); 
		q.pop();
        for(auto it:g[x])
		{
            y=it.first; k=it.second;
            if(dis[y]>dis[x]+k)
			{
                dis[y]=dis[x]+k;
                q.push(make_pair(dis[y],y));
            }
        }
    }
}
vector<int> q[2010];
int aaa[2010][2010];
int main() 
{
	int n,m,x,y,z;
	int i,j,k;
	scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++)
	{
        scanf("%d%d%d",&x,&y,&k);
        g[x].push_back(make_pair(y,k));
    }
    memset(w,0x3f,sizeof(w));
    for(int i=1;i<=n;i++)
	dij(i,w[i]);
    for(i=1;i<=n;i++)
    for(auto gg:g[i])
    if(w[i][gg.first]==gg.second)
	{
        q[i].push_back(gg.first);
		aaa[i][gg.first]=1;
    }
    int ans=0;
    for(i=1;i<=n;i++)
    for(j=1;j<=n;j++)
	{
        if(aaa[i][j]||i==j||w[i][j]==inf)
		{
            ans++; 
			continue;
        }
        for(auto it:q[i])
		if(aaa[it][j]&&w[i][j]==w[i][it]+w[it][j])
		{
            ans++; q[i].push_back(j); 
			aaa[i][j]=1; 
			break;
        }
    }
    
    printf("%d\n",ans);
    return 0;
}