【模拟试题】德雷克的追捕

最新推荐文章于 2024-10-12 20:26:07 发布

hwzzyr

最新推荐文章于 2024-10-12 20:26:07 发布

阅读量345

点赞数

分类专栏：矩阵，快速幂文章标签：快速幂矩阵

本文链接：https://blog.csdn.net/hwzzyr/article/details/77170242

版权

矩阵，快速幂专栏收录该内容

0 篇文章 0 订阅

订阅专栏

【模拟试题】德雷克的追捕

Description

　　这一次，德雷克要追捕一个从古代遗迹中逃脱的超自然人造物。
　　在古代遗迹中，德雷克拿到了人造物的OS说明，凭借出题人给他开的挂，他看懂了。
　　人造物会严格地从输入的地图中的起点出发，总是走最短路；也就是说，它访问任何一个结点时，走的路线都是从古代遗迹到这里的最短路。同时，如果没有最短路可走，人造物会选择停下。为了简化题目，我们保证从古代遗迹（结点1）到任何一个结点的最短路都是唯一的。
　　在每个结点（包括古代遗迹），人造物都可以找到一套该文明留下的强化工具对自己进行强化。每个结点上的工具都有五个参量：mul_1, c, mul_2, plus, times；设人造物当前综合值为now，则强化一次之后，now = now * mul_1 + c, c = c * mul_2 + plus, times = times - 1，直到times = 0时停止强化。
　　为了准备对付人造物的工具，德雷克需要知道人造物在每个结点可能的综合值。为了方便出题人，这个值当然要Mod 998244353啦。

Input

　　第一行包括三个整数n, m, st，表示人造物地图中的结点数和边数，以及人造物最初的综合值。
　　接下来n行每行五个整数，按顺序给出题目描述中所说的五个参量。
　　再接下来m行每行三个整数u, v, w，表示地图中存在u, v间的双向道路，权值为w。

Output

　　包括n行，每行一个整数，为人造物在每个结点的综合值。

Sample Input

Sample Output

Hint

　　对于30%的数据，n <= 10, m <= 100, times <= 10；
　　对于60%的数据，n <= 1000, m <= 5000, times <= 100；
　　对于100%的数据，n <= 100000, m <= 500000, 1 <= times <= 1000000，所有数值均在[0,Mod)范围内，保证任意最短路径长不超过100000000；出题人保证图连通。

Source

liuyunhui123

Solution

首先感谢一下liuyunhui123 dalao给蒟蒻们出的题。
大意是给出一张图，求出最短路径树（数据保证唯一），然后在每一个节点上进行time[i]次操作，当前节点的值成为儿子的初值，输出每一个节点的值。
因为数据保证，所以我们直接求最短路，在过程中记录一下更新的来源节点，也就是树上的父亲。构造出树之后，直接暴力操作就有60分（良心啊）。
但是100%的数据中time[i]<=10^6,直接操作是会迎来TLE的。所以我们要想办法加快这个过程。法一是推数学公式（出题人是这样说的，ORZ，蒟蒻不会），法二就是构造矩阵然后快速幂。
(now, (mul_1,1,0
c, * 0,mul_2,plus
1,) 0,0,1)
(矩阵太丑了。。。。)

CODE

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const long long mod=998244353ll;
inline int read(){
	char c;int rec=0;
	while((c=getchar())<'0'||c>'9');
	while(c>='0'&&c<='9')rec=rec*10+c-'0',c=getchar();
	return rec;
}
int n,m,st;
struct Node {int mul1,c,mul2,p,t;}point[100005];
struct Road {int next,to,val;}road[1000005];
int h[100005],cnt=0;
inline void add(int x,int y,int z)
{road[++cnt].to=y;road[cnt].next=h[x];h[x]=cnt;road[cnt].val=z;return ;}

int dis[100005],vis[100005],prep[100005];
inline void Spfa(int st){
	int i,j,v;
	memset(dis,0x3f,sizeof(dis));
	queue<int> q;
	q.push(st);dis[st]=0;vis[st]=1;
	while(!q.empty()){
		v=q.front();q.pop();vis[v]=0;
		for(i=h[v];i;i=road[i].next){
			j=road[i].to;
			if(dis[j]>dis[v]+road[i].val){
				dis[j]=dis[v]+road[i].val;
				prep[j]=v;
				if(!vis[j]){q.push(j);vis[j]=1;}
			}
		}
	}return ;
}
struct Branch {int next,to;}branch[100005];
inline void Add(int x,int y)
    {branch[++cnt].to=y;branch[cnt].next=h[x];h[x]=cnt;return ;}
inline void ReBuild(){
	memset(h,0,sizeof(h));cnt=0;
	for(int i=1;i<=n;i++)
		if(prep[i])Add(prep[i],i);
	return ;
}
int maxx=0,F=0;
long long ans[100005];
struct Matrix {long long a[3][3];}A,B,C;
inline void Zero(Matrix &x){
	for(int i=0;i<=2;i++){
		for(int j=0;j<=2;j++){
			x.a[i][j]=0;
		}
	}
}
inline void One(Matrix &x){
	Zero(x);
	for(int i=0;i<=2;i++)x.a[i][i]=1ll;
	return ;
}
inline Matrix Mul(Matrix x,Matrix y){
	Matrix rec;
	for(int i=0;i<=2;i++){
		for(int j=0;j<=2;j++){
			rec.a[i][j]=0;
			for(int k=0;k<=2;k++){
				rec.a[i][j]+=x.a[i][k]*y.a[k][j];
				rec.a[i][j]%=mod;
			}
		}
	}
	return rec;
}
inline long long KSM(long long val,long long c,long long m1,long long m2,long long p,long long t){
	A.a[0][0]=m1;A.a[0][1]=0;A.a[0][2]=0;
    A.a[1][0]=1;A.a[1][1]=m2;A.a[1][2]=0;
    A.a[2][0]=0;A.a[2][1]=p;A.a[2][2]=1;
    One(B);
    while(t){
    	if(t&1)
    		B=Mul(B,A);
		A=Mul(A,A);t>>=1;
	}
	long long rec;
	rec=(val*B.a[0][0]%mod+c*B.a[1][0]%mod+B.a[2][0])%mod;
	return rec;
}
inline long long Cal(int val,int id){
	long long rec=val;
	long long m1=point[id].mul1,m2=point[id].mul2,C=point[id].c;
	long long P=point[id].p,T=point[id].t;
	return KSM(rec,C,m1,m2,P,T);
}
inline void Sov(){
	queue<long long> q;
	long long i,j,v,w;
	q.push(1);
	while(!q.empty()){
		v=q.front();q.pop();
		w=ans[prep[v]];
		ans[v]=Cal(w,v);
		for(i=h[v];i;i=branch[i].next)
		    {j=branch[i].to;q.push(j);}
	}return ;
}
int main(){
	n=read();m=read();st=read();
	for(int i=1;i<=n;i++){
		point[i].mul1=read();point[i].c=read();
		point[i].mul2=read();point[i].p=read();
		point[i].t=read();maxx=max(maxx,point[i].t);
	}
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),z=read();
		add(x,y,z);add(y,x,z);
	}Spfa(1);ReBuild();ans[0]=st;
	Sov();
	for(int i=1;i<=n;i++)
		cout<<ans[i]<<'\n';
	return 0;
}

(矩阵最后取值的时候差点把a[2][0]给忘了。。。下次干脆一开始就乘进去)