【NOI P模拟赛】欢乐豆（最短路，线段树）

最新推荐文章于 2022-06-28 21:17:35 发布

DD(XYX)

最新推荐文章于 2022-06-28 21:17:35 发布

阅读量247

点赞数

分类专栏：线段树图论文章标签：图论

本文链接：https://blog.csdn.net/weixin_43960414/article/details/121248493

版权

线段树同时被 2 个专栏收录

53 篇文章 0 订阅

订阅专栏

图论

48 篇文章 0 订阅

订阅专栏

题面

在这里插入图片描述

样例输入

样例输出

数据范围

$1\leq n\leq10^5,1\leq m\leq 3000,0\leq a_i\leq10^6.$

题解

原图相当于每个点有个固定的代价可以跳到其他任一点，所以原图上的点，以及修改后的与修改边无关的点 $u, v$ 的最短路就是 $a_u$ 。

所以，除了修改边覆盖的点，其他的点的贡献都是 $a_i(n-1)$ 。

被修改边覆盖的点只有最多 $2 m$ 个，我们可以单独算这些点的贡献，暂时称之为特殊点。

对于一个特殊点 $i$ 和所有非特殊点 $j$ ， $d i s (i, j)$ 一定是个定值，因为所有非特殊点对于某个特殊点是等价的——不难发现它们一定是经过一段修改路径，再“跳”到目标的，通过跳一定可以到达所有非特殊点。

所以我们需要讨论的最短路也从 $2m\times n$ 对变成了 $2m\times 2m$ 对，难点在于计算特殊点之间的最短路。

特殊点之间的修改边有 $m$ 条，而“跳”边覆盖了连续的区间，因此，不难想到用线段树优化 Dijkstra （不是优化建图，是直接优化松弛）。枚举每个起点做单源最短路。

时间复杂度 $O(m^2\log m)$ ，需要卡常和剪枝。

CODE

#include<set>
#include<map>
#include<queue>
#include<stack>
#include<random>
#include<vector>
#include<bitset>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 100005
#define LL long long
#define ULL unsigned long long
#define DB double
#define lowbit(x) (-(x) & (x))
#define ENDL putchar('\n')
#define FI first
#define SE second
int xchar() {
	static const int maxn = 1000000;
	static char b[maxn];
	static int pos = 0,len = 0;
	if(pos == len) pos = 0,len = fread(b,1,maxn,stdin);
	if(pos == len) return -1;
	return b[pos ++];
}
//#define getchar xchar
LL read() {
    LL f=1,x=0;int s = getchar(); 
    while(s < '0' || s > '9') {if(s<0)return -1;if(s=='-')f=-f;s = getchar();}
    while(s >= '0' && s <= '9') {x = (x<<3) + (x<<1) + (s^48); s = getchar();}
    return f*x;
}
void putpos(LL x) {if(!x)return ;putpos(x/10);putchar('0'+(x%10));}
void putnum(LL x) {
    if(!x) {putchar('0');return ;}
    if(x<0) {putchar('-');x = -x;}
    return putpos(x);
}
void AIput(LL x,int c) {putnum(x);putchar(c);}

int n,m,s,o,k;
int Max(int a,int b) {return a>b ? a:b;}
int Min(int a,int b) {return a<b ? a:b;}
int a[MAXN];
map<int,int> G[MAXN];
vector<pair<int,int> > g[MAXN];
bool f[MAXN];
int b[MAXN],cn,rk[MAXN];
int dp[MAXN];
struct it{
	int nm,id;
	it(){nm=id=0;}
	it(int N,int I) {nm=N;id=I;}
}tre[MAXN];
int lz[MAXN],M;
void MN(it &y,int mn) {
	y.nm = Min(y.nm,mn);
}
it merg(it a,it b) {
	if(!a.id) return b;
	if(!b.id) return a;
	return a.nm < b.nm ? a:b;
}
void maketree(int n,int nb) {
	M=1;while(M<n+2)M<<=1;
	for(int i = 1;i < (M<<1);i ++) lz[i] = 0x3f3f3f3f,tre[i] = it();
	for(int i = 1;i <= n;i ++) tre[M+i].nm = nb,tre[M+i].id = i;
	for(int i = M-1;i > 0;i --) tre[i] = merg(tre[i<<1],tre[i<<1|1]);
}
void addtree(int x,it y) {
	int s=M+x; tre[s] = y; s >>= 1;
	while(s) MN(tre[s] = merg(tre[s<<1],tre[s<<1|1]),lz[s]),s >>= 1;
}
void addsg(int l,int r,int y) {
	if(l > r) return ;
	for(int s=M+l-1,t=M+r+1;s||t;s>>=1,t>>=1) {
		if(s<M) MN(tre[s] = merg(tre[s<<1],tre[s<<1|1]),lz[s]);
		if(t<M) MN(tre[t] = merg(tre[t<<1],tre[t<<1|1]),lz[t]);
		if((s>>1) ^ (t>>1)) {
			if(!(s&1)) MN(tre[s^1],y),lz[s^1] = Min(lz[s^1],y);
			if(t & 1) MN(tre[t^1],y),lz[t^1] = Min(lz[t^1],y);
		}
	}return ;
}
int main() {
	freopen("happybean.in","r",stdin);
	freopen("happybean.out","w",stdout);
	n = read();m = read();
	LL ans = 0;
	for(int i = 1;i <= n;i ++) {
		a[i] = read();
		ans += a[i]*1ll*(n-1);
	}
	for(int i = 1;i <= m;i ++) {
		s = read();o = read();k = read();
		G[s][o] = k;
		f[s] = 1;f[o] = 1;
	}
	int mn = 0x3f3f3f3f;
	for(int i = 1;i <= n;i ++) {
		if(!f[i]) {
			mn = min(mn,a[i]);
		}
		else {
			ans -= a[i]*1ll*(n-1);
			b[++ cn] = i;rk[i] = cn;
			for(auto j = G[i].begin();j != G[i].end();j ++) {
				g[cn].push_back(*j);
			}
		}
	}
	for(int i = 1;i <= cn;i ++) {
		int me = a[b[i]];
		if(g[i].empty()) {
			ans += a[b[i]]*1ll*(n-1);
			continue;
		}
		maketree(cn,Min(0x3f3f3f3f,a[b[i]]+mn));
		addtree(i,it(0,i));
		for(int ii = 1;ii <= cn;ii ++) {
			int t = tre[1].id;
			if(!t) break;
			dp[t] = tre[1].nm;
			ans += dp[t];
			me = Min(me,dp[t] + a[b[t]]);
			addtree(t,it());
			int nmn = min((LL)0x3f3f3f3f,0ll+dp[t]+a[b[t]]+mn);
			MN(tre[1],nmn);
			lz[1] = Min(lz[1],nmn);
			int pr = 0;
			for(int j = 0;j < (int)g[t].size();j ++) {
				int y = rk[g[t][j].FI],w = g[t][j].SE;
				addsg(pr+1,y-1,dp[t] + a[b[t]]);
				pr = y; addsg(y,y,dp[t]+w);
			}
			addsg(pr+1,cn,dp[t] + a[b[t]]);
		}
		ans += me*1ll*(n-cn);
	}
	AIput(ans,'\n');
	return 0;
}