CF773D Perishable Roads

思路:

注意到答案应该是链加上一串贡献相同的树的贡献,因为若 \(a \to u\) 的贡献 比 \(b \to u\) 的贡献小,那么可以连 \(b \to a\),答案会更优。

那么有一个贪心思路,对于每个根,找到连向这个根的最短边,然后对于这条边的另一个端点,也找到连向这个端点的最短边,以此类推;很显然,这个假了。

\(T\) 为当前根,考虑找到全局最短边 \((u,v,w)\),考虑令 \(u \to T\),然后其它所有点都连 \(v\),这样其它点到 \(T\) 的贡献必然是最小的,但是若 \(u \to T\) 的贡献非常大,那这样也是不优的。

则考虑组成一个 \(u \to T\) 的一条链,使得这条链的贡献加上其它所有点的贡献最优,设 \(d_T\) 表示从 \(u\)\(T\) 组成的链的贡献的最小值,\(x\)\(u \to T\)\(T\) 以外的点的个数。

则贡献为 \(d_T + (n-x-1) w\),发现这个 \(x\) 很烦,考虑去掉,即我们计算 \(d\) 的时候,对于全局边都减去一个 \(w\),则 \(d_T' = d_T - x \times w\) 了。

此时贡献为 \(d_T + (n-1)w\),则我们要求出 \(d_T\) 的最小值,发现是个单源最短路问题,直接跑 dijkstra 即可。

时间复杂度为 \(O(N^2)\)

完整代码:

#include<bits/stdc++.h>
#define Add(x,y) (x+y>=mod)?(x+y-mod):(x+y)
#define lowbit(x) x&(-x)
#define pi pair<ll,ll>
#define pii pair<ll,pair<ll,ll>>
#define iip pair<pair<ll,ll>,ll>
#define ppii pair<pair<ll,ll>,pair<ll,ll>>
#define fi first
#define se second
#define full(l,r,x) for(auto it=l;it!=r;it++) (*it)=x
#define Full(a) memset(a,0,sizeof(a))
#define open(s1,s2) freopen(s1,"r",stdin),freopen(s2,"w",stdout);
#define For(i,l,r) for(register int i=l;i<=r;i++)
#define _For(i,l,r) for(register int i=r;i>=l;i--)
using namespace std;
typedef double db;
typedef unsigned long long ull;
typedef long long ll;
bool Begin;
const ll N=2020,INF=1e18;
inline ll read(){
    ll x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-')
          f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=(x<<1)+(x<<3)+(c^48);
        c=getchar();
    }
    return x*f;
}
inline void write(ll x){
	if(x<0){
		putchar('-');
		x=-x;
	}
	if(x>9)
	  write(x/10);
	putchar(x%10+'0');
}
ll n,w,Min=INF,id;
ll a[N][N],d[N];
bool f[N];
void dijkstra(ll s){
	f[s]=1;
	For(i,1,n){
		d[i]=a[s][i];
		For(j,1,n){
			if(i==j)
			  continue;
			d[i]=min(d[i],a[i][j]*2);
		}
	}
	For(i,1,n-1){
		Min=INF,id=0;
		For(j,1,n){
			if(f[j])
			  continue;
			if(d[j]<Min){
				Min=d[j];
				id=j;
			}
		}
		f[id]=1;
		For(j,1,n){
			if(f[j])
			  continue;
			d[j]=min(d[j],d[id]+a[id][j]);
		}
	}
}
bool End;
int main(){
//	open("A.in","A.out");
	n=read();
	For(i,1,n){
	    For(j,i+1,n){
	    	a[i][j]=a[j][i]=read();
	    	if(a[i][j]<Min){
	    		Min=a[i][j];
	    		id=i;
			}
		}
	}
	For(i,1,n)
	  For(j,1,n)
		if(a[i][j])
		  a[i][j]-=Min;
	w=Min;
	dijkstra(id);
	For(i,1,n){
		write(d[i]+(n-1)*w);
		putchar('\n');
	}
	cerr<<'\n'<<abs(&Begin-&End)/1048576<<"MB";
	return 0;
}