[二次建图] [拆点] 模拟赛好题分享

题目信息很复杂，修改颜色可能会对之后的路径产生影响，但我们会发现一点：边数 = 颜色数

就是说，除了一个点上连 $M$ 条边以外（菊花），一个修改操作一定能将这条边修改为与其他完全不同的颜色，假设从 $u$ 走到 $v$，$v$ 接下来的选择就不再会被这条边所限制

由于是最短路，显然一条边只会经过一次，那么只需要知道对于一条边 $u\to v$ 的花费有哪几种，建边后就可以直接跑最短路了

1. 直接修改 $u_i\to v_i$，花费 $P_i$
2. 修改与这条边颜色相同的、$u_i$ 的所有出边，随后直接走到 $v_i$，花费 $Sum[u_i][c]-P(u_i,v_i)$

直接这样建是否就够了呢？通过手造数据我们发现有一种情况：

假设需要 $u\to v\to w$，  $u\to v$ 时走颜色 $c$ 进行了一次操作 1 ，$v\to w$ 时 如果走 $c$ 且进行操作 2，那么支付 2. 的代价显然是多了，因为第一次操作已经减少一条 颜色 $c$ 的边

dj 的队列中无法记录这个信息，那么我们只能通过 二次建图 来额外对距离进行限制

最朴素的做法，显然是对所有上面那种情况的 $u\to w$ 直接连一条边，权值为 $Sum[v][c]-val(v,w)$，这样的复杂度是不可接受的

考虑简化：

对于 后半段的一条边 $(v,w,c)$ ，不管怎样的 $(u,v,c)$ ，拼起来后的 $(u,w)$ 总花费都是一样的

这就在启发我们将 权值全部放在后半段上，前半段权值为 $0$ ，代表可以随意到这个点，出去时再计算总代价

发现限制只与颜色有关，考虑对 $v$ 拆点，每种颜色一个点，得到下图：

这是对于 连接红色边的 $v$ 的情况，建立新点 $v_{red}$，与其他原图上的点相连

对于一个点，它有几种颜色的出边，就拆成几个点，故总点数为 $N+2M$

原图中一条 1操作中 $u\to v$ 的有向边会被 在拆点后的图上建两次，加上 1、2 操作本身要建的边，总边数为 $8M$

直接跑 dj 可过

#include<bits/stdc++.h>
using namespace std; 

typedef long long LL ;
const int N = 1e5 + 100 ;

int n , m , R ;
map<int,int> mp[N] ;
map<int,LL> Sum[N] ;
struct nn
{
	int lst , to , c ;
	LL val ;
}E[16*N] ; // 8m 条边 
int head[5*N] , tot ; // n+2m 个点 
inline void add( int x , int y , int c , LL p ) 
{
	E[++tot] = (nn){ head[x] , y , c , p } ;
	head[x] = tot ;
}

bool vis[5*N] ;
LL dis[5*N] ;
struct nod
{
	int id ; LL val ;
	friend bool operator < ( nod x , nod y ) {
		return x.val > y.val ;
	}
};
priority_queue<nod> q ;
void dij()
{
	memset( dis , 0x3f , sizeof dis ) ;
	dis[1] = 0 ;
	q.push( (nod){1,0} ) ;
	while( !q.empty() ) {
		int x = q.top().id ; q.pop() ;
		if( vis[x] ) continue ;
		vis[x] = 1 ;
		for(int i = head[x] ; i ; i = E[i].lst ) {
			int t = E[i].to ;
			if( dis[t] > dis[x] + E[i].val ) {
				dis[t] = dis[x] + E[i].val ;
				q.push( (nod){t,dis[t]} ) ;
			}
		}
	}
}

int main() 
{
	scanf("%d%d" , &n , &m ) ;
	R = n ;
	int a , b , c , p ;
	for(int i = 1 ; i <= m ; i ++ ) {
		scanf("%d%d%d%d" , &a , &b , &c , &p ) ;
		add( a , b , c , p ) , add( b , a , c , p ) ; // 操作 1 的 
		if( mp[a].find(c) == mp[a].end() ) {
			mp[a][c] = ++R ;
		}
		Sum[a][c] += p ;
		if( mp[b].find(c) == mp[b].end() ) {
			mp[b][c] = ++R ;
		}
		Sum[b][c] += p ;
	}
	for(int i = 1 ; i <= n ; i ++ ) {
		for(int j = head[i] ; j ; j = E[j].lst ) {
			int t = E[j].to ;
			if( E[j].c == 0 ) continue ; // 不是操作 1 的就不管了 
			add( i , mp[t][E[j].c] , 0 , 0 ) ;
			add( mp[i][E[j].c] , t , 0 , Sum[i][E[j].c]-E[j].val ) ;//新加的 
			add( i , t , 0 , Sum[i][E[j].c]-E[j].val ) ;// 操作 2 的 
		}
	}
	dij() ;
	printf("%lld\n" , dis[n]>1e17?-1:dis[n] ) ;
	return 0;
}