考试
T1 toy 100 真正要计算东南西北的只有len/n后再len%n的一部分。算是普及组T1难度这种吧,直接跳过。
T2 running 10 说来很可惜,已经差不多正解了,还是因为没想清楚所以wa了一部分。不过求lca的方法不是倍增可能也是一部分失分原因,归根结底还是懒。
我算除了每个节点的深度、子节点个数,然后每次查询的两个点算他们的距离和lca,但是最后分类讨论求结果的时候搞错了,加上时间紧迫有点慌。
我考试时写的是:【(21:00)更新1:这个是错的,正解等下打,打好会更新】【(22:03)更新2:还是只能过第1个点,暂时还未找出错误。】
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn (100000+10)
int n, m, dis, sum, cnt, lca, bj;
int head[maxn], p[maxn], vis[maxn], deep[maxn], son[maxn];
struct node{
int to, next;
}e[maxn];
template <typename T>
T read(){
T N(0), F(1);
char ch = getchar();
for(; !isdigit(ch); ch = getchar()) if(ch == '-') F = -1;
for(; isdigit(ch); ch = getchar()) N = N*10+ch-48;
return N*F;
}
void add(int x,int y){
e[++cnt] = (node){y, head[x]};
head[x] = cnt;
}
void dfs(int u){
for(int i = head[u]; i; i = e[i].next){
int v = e[i].to;
if(!deep[v]){
deep[v] = deep[u] + 1;
p[v] = u;
dfs(v);
}
}
}
void dist(int x, int y){
int dpx = deep[x]; int nx = x;
int dpy = deep[y]; int ny = y;
// printf("dpx = %d dpy = %d\n", dpx, dpy);
while(dpx > dpy){
dis++; dpx--;
nx = p[nx];
}
while(nx != ny){
dis += 2;
nx = p[nx];
ny = p[ny];
}
lca = nx;
}
int main(){
freopen("running.in", "r", stdin);
freopen("running.out","w",stdout);
n = read<int>();
for(int i = 1; i < n; i++){
int x, y;
x = read<int>(), y = read<int>();
add(x, y);
add(y, x);
}
deep[1] = 1;
dfs(1);
son[1] = n;
for(int i = 2; i <= n; i++){
int j = i;
while(j != 1){
son[j]++;
j = p[j];
}
}
m = read<int>();
for(int i = 1; i <= m; i++){
int x, y;
x = read<int>(); y = read<int>();
if(x == y){
printf("%d\n", n);
continue;
}
if(deep[x] < deep[y]) swap(x, y);
dis = 0;
dist(x, y);
// printf("dis = %d lca = %d\n", dis, lca);
if(dis&1){
printf("0\n");
continue;
}
if(deep[x] == deep[y]){
int nx = x; int dp = deep[x];
int ny = y;
while(dp > deep[lca]+1){
dp--;
nx = p[nx];
ny = p[ny];
}
printf("%d\n", n - son[nx] - son[ny]);
continue;
}
int s = dis/2;
int nx = x;
while(s--){
nx = p[nx];
if(s == 1) bj = nx;
}
printf("%d %d %d %d\n",deep[nx], deep[bj], son[nx], son[bj]);
printf("%d\n", son[nx] - son[bj]);
}
return 0;
}
以 1 为根,求出每个点的深度
depi
和子树大小
sizei
[我的代码里是son],设两人所在点 x,y 的 LCA 为 z[我代码里写的是lca],两点之间的距离为 d。
分下列几种情况:
1、当 d 为奇数时,答案为 0(我写的d是点数所以是偶数时答案为0)
2、否则,若 x=y,答案为 n(这个判断了)
3、否则,若
depx=depy
,令 p1 为 x 的祖先,且
depp1=depz+1
,令 p2 为 y 的祖先,且
depp2=depz+1
,则答案为
n−sizep1−sizep2
。
4、否则,假设
depx>depy
,令 p1 为 x 的祖先,且
depp1=depx−d/2
且
depp2=depp1+1
,则答案为
sizep1−sizep2
发现我总是分类讨论扯不清,思路总是不能全面只朝一个方向思考。
T3 classroom 20
本来写的50分做法【预处理阶乘和阶乘的逆元,然后逐一计算组合数。】,结果数组开小了,有几个地方ll写成了int!!!!!!下次要注意!!!!
正解的话是循环矩乘,基本上懂了,但是关于这类题目下一次我还是无法写出递推式,更不会从递推式到做出矩乘优化。
以下是改完的程序。时限1.5s
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn (1500+10)
ll n, k, p, r;
ll C[maxn];
template <typename T>
T read(){
T N(0), F(1);
char ch = getchar();
for(; !isdigit(ch); ch = getchar()) if(ch == '-') F = -1;
for(; isdigit(ch); ch = getchar()) N = N*10 + ch-48;
return N*F;
}
void mul(ll *A, ll *B, ll q){
for(int i = 0; i <= 2*q; i++) C[i] = 0LL;
for(int i = 0; i < q; i++){
for(int j = 0; j < q; j++){
C[i+j] = (C[i+j]%p + (A[i]%p*B[j]%p))%p;
}
}
for(int i = q; i <= 2*q-2; i++) C[i-q] = (C[i-q]%p + C[i]%p)%p;
for(int i = 0; i <= q; i++) A[i] = C[i];
}
void pow(ll *A, ll *ret, ll u, ll v){
for(int i = 0; i <= 2*v; i++) ret[i] = 0LL;
ret[0] = 1LL;
while(u){
if(u&1LL) mul(ret, A, v);
mul(A, A, v);
u >>= 1LL;
}
}
ll A[maxn], B[maxn];
int main(){
freopen("class.in", "r", stdin);
freopen("class.out","w",stdout);
n = read<ll>(); p = read<ll>(); k = read<ll>(); r = read<ll>();
A[0] = A[1] = 1;
if(k == 1) A[1] = 0, A[0] = (A[0]+1)%p;//特判这里没太弄明白,知道要在k==1、r==0时特判但是这个写法不太明白。
pow(A, B, n*k, k);
printf("%lld\n", B[r]);
return 0;
}
今天算是结束了,倍增最近考的多,要练练手!