排位赛（5）“发愤”补题

C

题面：

题目大意：

给定一棵有n个节点的树，选择一个点作为首都，有q个询问，询问1是改变首都，询问2是问所有点到首都会有多少经过询问节点。
数据范围：1 <= N <= 1e5, 1 <= Q <= 5e4

思路：

答案明显和一个点的子树大小有关。
暴力的方法我们会想每更新一次首都都dfs一次。甚至乎比赛过程中，我还脑洞了一种方法：dfs一次把所有点的所有边连出去的树大小求出，询问时，dfs判断首都在询问节点(设为A)的哪一支（设为B），答案是n-sz[A][B]+1。但是…明显超时(/ω＼)

正确的做法：

以1为根节点，dfs过程中我们把每个点的深度(dep[])、子树大小(sz[])和父节点（f[][0]）求出。
设首都为r,询问节点为d。
若dep[r] <= dep[d]，无论节点之间是什么关系,ans = sz[d]；
若dep[r] > dep[d]，即首都节点在下面，这时可能有2种情况：
1、首都节点不为询问节点的子节点，ans = sz[d]；
2、首都节点为询问节点的子节点，此时设d含首都节点的子树中离最近的点为k，答案应为 ans = n - sz[k]，而用树上倍增的方法求出k，和求LCA差不多。

AC代码

#include<bits/stdc++.h>
#define pb push_back
#define lowbit(x) x&(-x)
#define mst(a,b) memset(a,b,sizeof a)
#define fi first
#define se second
#define mp make_pair

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int INF = 0x3f3f3f3f;
const double eps = 1e-9;
const int Mod = 1e9+7;
const int N = 1e5+10;

ll Read(){
	ll val = 0, opt = 1; char ch;
	while (!isdigit(ch = getchar())) if (ch == '-') opt = -1;
	while (isdigit( ch )) (val *= 10) += ch - '0', ch = getchar();
	return val * opt;
}

int n,q,r;
vector<int> g[N];
int dep[N],sz[N];	//深度、子树大小
int f[N][30];
 
void ini(){
	for(int i=1;i<=n;i++){
		sz[i]=1;
		g[i].clear();
	}
	mst(dep, 0);
	mst(f, -1);
}
 
void dfs(int u, int fa){
	dep[u] = dep[fa] + 1;
	f[u][0] = fa;
	for(auto i : g[u]){
		if(i == fa) continue;
		dfs(i, u);
		sz[u]+=sz[i];
	}
}

void solve(){	
	n = Read(); q = Read(); r = Read();
	ini();
	
	for(int i=0;i<n-1;i++){
		int x,y;
		x = Read(); y = Read();
		g[x].pb(y);
		g[y].pb(x);
	}
	dfs(1,0);
	//倍增数组
	for(int i=0;i<25;i++) {		//2^25 > 3e7
		for(int j=1;j<=n;j++){
			if(f[j][i] != -1) f[j][i+1] = f[ f[j][i] ][i];
		}
	}
	//询问
	while(q--){
		int c,d;
		c=Read();d=Read();
		if(c == 0){
			r = d; 
		}
		else{
			if(r == d) printf("%d\n", n);
			else if(dep[r] <= dep[d]){
				printf("%d\n",sz[d]);
			}
			else{
				int jp = r;
				for(int i=25;i>=0;i--){
					if(f[jp][i] != -1 && dep[f[jp][i]] > dep[d]){
						jp = f[jp][i];
					}
				}
				if(f[jp][0] == d){
					printf("%d\n", n-sz[jp]);
				}
				else printf("%d\n", sz[d]);
			}
		}
	} 
}

signed main(){
	int cas;
	cas = Read();
	for(int i=1;i<=cas;i++){
		printf("Case #%d:\n",i);
		solve();
	}
    return 0;
}

I

题面：

题目大意：

求斐波那契数列（1,1,2,3,5,8…）的第（2+k*a）项(k = 0,1,2,3,4,…)的和。
数据范围：非常大，10^18级别

思路：

矩阵快速幂

这里f[1] = 2, f[2] = 3, …
初始状态定为（0,1,1）

AC代码

#include<bits/stdc++.h>
#define pb push_back
#define lowbit(x) x&(-x)
#define mst(a,b) memset(a,b,sizeof a)
#define fi first
#define se second
#define mp make_pair

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int INF= 0x3f3f3f3f;
const double eps = 1e-9;
const int Mod = 1e9+7;
const int N = 1e5+10;

struct mat{
	int n;
	vector<vector<long long> > val;
	//构造函数 
	explicit mat(int sz) : val(sz,vector<long long>(sz)){
		n=sz;
	}
	//单位矩阵 
	void emat(){
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				val[i][j]= (i==j);
			}
		}
	}
	//重载乘号
	mat operator * (const mat &a) const{
		mat res(n);
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				for(int k=0;k<n;k++){
					res.val[i][j]=(res.val[i][j]+val[i][k]*a.val[k][j])%Mod;
				}
			}
		}
		return res;
	}  
}; 

//快速幂
mat matqp(mat a,long long b){
	mat res(a.n);
	res.emat();
	while(b){
		if(b&1) res=res*a;
		a=a*a;
		b>>=1;
	}
	return res;
}

signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(0);
	
	int cas;
	cin>>cas;
	int tt=1;
	
	while(cas--){
		ll l,r,k;
		cin>>l>>r>>k;
		
		mat trans(2);
		trans.val[0][0]=1;
		trans.val[0][1]=1;
		trans.val[1][0]=1;
		trans.val[1][1]=0;
		mat res=matqp(trans,k);
		
		mat p(3);
		p.val[0][0]=1;p.val[0][1]=res.val[0][0];p.val[0][2]=res.val[0][1];
		p.val[1][0]=0;p.val[1][1]=res.val[0][0];p.val[1][2]=res.val[0][1];
		p.val[2][0]=0;p.val[2][1]=res.val[1][0];p.val[2][2]=res.val[1][1];
		
		l-=1;
		mat r1=matqp(p,(r/k));
		mat r2=matqp(p,(l/k));

		ll ans=r1.val[0][2]+r1.val[0][1]-r2.val[0][2]-r2.val[0][1];
		cout<<"Case "<<tt<<": "<<(ans+2*Mod)%Mod<<endl;	//负数取模 
		tt++;
	} 
    return 0;
}

细节：

1、确定初始状态的时候花了好多时间，后来才发现不是从第一项开始的┭┮﹏┭┮
2、矩阵快速幂的板子有个细节问题，有个地方使用了引用改变了原矩阵

mat matqp(mat &a,long long b){
mat matqp(mat a,long long b){

3、这里ans可能是负数的，取模前先+k*Mod

ll ans=r1.val[0][2]+r1.val[0][1]-r2.val[0][2]-r2.val[0][1];
cout<<"Case “<<tt<<”: "<<(ans+2*Mod)%Mod<<endl;