一、AcWing 1050. 鸣人的影分身
【题目描述】
在火影忍者的世界里,令敌人捉摸不透是非常关键的。
我们的主角漩涡鸣人所拥有的一个招数:多重影分身之术,就是一个很好的例子。
影分身是由鸣人身体的查克拉能量制造的,使用的查克拉越多,制造出的影分身越强。
针对不同的作战情况,鸣人可以选择制造出各种强度的影分身,有的用来佯攻,有的用来发起致命一击。
那么问题来了,假设鸣人的查克拉能量为MMM,他影分身的个数为NNN,那么制造影分身时有多少种不同的分配方法?
注意:
- 影分身可以分配000点能量。
- 分配方案不考虑顺序,例如:M=7,N=3M=7,N=3M=7,N=3,那么(2,2,3)(2,2,3)(2,2,3)和(2,3,2)(2,3,2)(2,3,2)被视为同一种方案。
【输入格式】
第一行是测试数据的数目ttt。
以下每行均包含二个整数M,NM,NM,N,以空格分开。
【输出格式】
对输入的每组数据M,NM,NM,N,用一行输出分配的方法数。
【数据范围】
0≤t≤200≤t≤200≤t≤20
1≤M,N≤101≤M,N≤101≤M,N≤10
【输入样例】
1
7 3
【输出样例】
8
【分析】
状态表示:f[i][j]f[i][j]f[i][j]表示总和为iii且恰好表示成jjj个数的和的方案数。
状态计算:
- 当前的jjj个数中的最小值为000,那么我们将这个000删去,则总和不变,元素数量减一,即f[i][j]=f[i][j−1]f[i][j]=f[i][j-1]f[i][j]=f[i][j−1];
- 当前的jjj个数中的最小值大于000,那么我们将每个数减111,则总和减jjj,元素数量不变,即f[i][j]=f[i−j][j]f[i][j]=f[i-j][j]f[i][j]=f[i−j][j]。
由于元素数量固定,因此最后的答案即为f[m][n]f[m][n]f[m][n]。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 20;
int f[N][N];//f[i][j]表示总和为i且恰好表示成j个数的和的方案
int n, m;
int main()
{
int T;
cin >> T;
while (T--)
{
cin >> m >> n;
f[0][0] = 1;
for (int i = 0; i <= m; i++)
for (int j = 1; j <= n; j++)
{
f[i][j] = f[i][j - 1];//集合中的最小值为0
if (i >= j) f[i][j] += f[i - j][j];//集合中的最小值大于0
}
cout << f[m][n] << endl;
}
return 0;
}
二、AcWing 1047. 糖果
【题目描述】
由于在维护世界和平的事务中做出巨大贡献,Dzx被赠予糖果公司2010年5月23日当天无限量糖果免费优惠券。
在这一天,Dzx可以从糖果公司的NNN件产品中任意选择若干件带回家享用。
糖果公司的NNN件产品每件都包含数量不同的糖果。
Dzx希望他选择的产品包含的糖果总数是KKK的整数倍,这样他才能平均地将糖果分给帮助他维护世界和平的伙伴们。
当然,在满足这一条件的基础上,糖果总数越多越好。
Dzx最多能带走多少糖果呢?
注意:Dzx只能将糖果公司的产品整件带走。
【输入格式】
第一行包含两个整数NNN和KKK。
以下NNN行每行111个整数,表示糖果公司该件产品中包含的糖果数目,不超过100000010000001000000。
【输出格式】
符合要求的最多能达到的糖果总数,如果不能达到KKK的倍数这一要求,输出000。
【数据范围】
1≤N≤1001≤N≤1001≤N≤100
1≤K≤1001≤K≤1001≤K≤100
【输入样例】
5 7
1
2
3
4
5
【输出样例】
14
【样例解释】
Dzx的选择是2+3+4+5=142+3+4+5=142+3+4+5=14,这样糖果总数是777的倍数,并且是总数最多的选择。
【分析】
状态表示:f[i][j]f[i][j]f[i][j]表示所有只从前iii个物品中选,且糖果总和除以kkk的余数为jjj的所有方案的最大糖果数。
状态计算:
- 不选第iii个物品:f[i][j]=f[i−1][j]f[i][j]=f[i-1][j]f[i][j]=f[i−1][j];
- 选第iii个物品:那么前i−1i-1i−1个物品的余数应该为(j−wi)%k(j-w_i)\% k(j−wi)%k,即f[i][j]=f[i−1][((j−wi)%k+k)%k]f[i][j]=f[i-1][((j-w_i)\% k+k)\% k]f[i][j]=f[i−1][((j−wi)%k+k)%k]。
注意除了f[0][0]f[0][0]f[0][0]合法之外其他所有方案都是不合法的,因此除了f[0][0]f[0][0]f[0][0]初始化为000以外其余的fff都初始化为负无穷。最后的答案即为f[n][0]f[n][0]f[n][0]。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110;
int f[N][N];//f[i][j]表示所有只考虑前i个物品,且价值总和模k的余数为j的方案的最大值
int n, k;
int main()
{
cin >> n >> k;
memset(f, 0x8f, sizeof f);
f[0][0] = 0;
for (int i = 1; i <= n; i++)
{
int w;
cin >> w;
for (int j = 0; j < k; j++)
f[i][j] = max(f[i - 1][j], f[i - 1][((j - w) % k + k) % k] + w);
}
cout << f[n][0] << endl;
return 0;
}
三、AcWing 1222. 密码脱落
【题目描述】
X星球的考古学家发现了一批古代留下来的密码。
这些密码是由A,B,C,DA,B,C,DA,B,C,D四种植物的种子串成的序列。
仔细分析发现,这些密码串当初应该是前后对称的(也就是我们说的镜像串)。
由于年代久远,其中许多种子脱落了,因而可能会失去镜像的特征。
你的任务是:给定一个现在看到的密码串,计算一下从当初的状态,它要至少脱落多少个种子,才可能会变成现在的样子。
【输入格式】
共一行,包含一个由大写字母A,B,C,DA,B,C,DA,B,C,D构成的字符串,表示现在看到的密码串。
【输出格式】
输出一个整数,表示至少脱落了多少个种子。
【数据范围】
输入字符串长度不超过100010001000。
【输入样例1】
ABCBA
【输出样例1】
0
【输入样例2】
ABDCDCBABC
【输出样例2】
3
【分析】
首先我们在原字符串中找出最长的回文子序列,如下图所示,其中红色标出的即为最长回文子序列:
对于所有不在最长回文子序列中的字符(蓝色标记的字符),我们都可以在其对应的位置插入同样的字符使其匹配(绿色标记的字符),那么从当前字符串变成初始状态需要添加叶子的数量就等价于当前字符串变成最长回文子序列需要减去的叶子数量。即至少脱落多少个种子等价于原序列的长度减去最长回文子序列的长度。
状态表示:f[i][j]f[i][j]f[i][j]表示所有在区间[i,j][i,j][i,j]中的回文子序列的最大长度。
状态计算:
- 若区间[i,j][i,j][i,j]中的回文子序列包含s[i]s[i]s[i]和s[j]s[j]s[j],也就是满足s[i]=s[j]s[i]=s[j]s[i]=s[j]时,则区间[i,j][i,j][i,j]中的回文子序列最大长度就为区间[i+1,j−1][i+1,j-1][i+1,j−1]中的回文子序列最大长度再加上这两个字符,即f[i][j]=f[i+1][j−1]+2f[i][j]=f[i+1][j-1]+2f[i][j]=f[i+1][j−1]+2;
- 若不包含s[i]s[i]s[i],则区间[i,j][i,j][i,j]中的回文子序列最大长度就为区间[i+1,j][i+1,j][i+1,j]中的回文子序列最大长度,即f[i][j]=f[i+1][j]f[i][j]=f[i+1][j]f[i][j]=f[i+1][j];
- 若不包含s[j]s[j]s[j],则区间[i,j][i,j][i,j]中的回文子序列最大长度就为区间[i,j−1][i,j-1][i,j−1]中的回文子序列最大长度,即f[i][j]=f[i][j−1]f[i][j]=f[i][j-1]f[i][j]=f[i][j−1]。
初始化:当区间长度为111时,每个字符自身为回文子序列,即f[i][j]=1f[i][j]=1f[i][j]=1。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
char s[N];
int f[N][N];
int main()
{
cin >> s;
int n = strlen(s);
for (int len = 1; len <= n; len++)
{
for (int i = 0; i < n; i++)
{
int j = i + len - 1;
if (len == 1) { f[i][j] = 1; continue; }
if (s[i] == s[j]) f[i][j] = f[i + 1][j - 1] + 2;
f[i][j] = max(f[i][j], max(f[i + 1][j], f[i][j - 1]));
}
}
cout << n - f[0][n - 1] << endl;
return 0;
}
四、AcWing 1220. 生命之树
【题目描述】
在X森林里,上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集SSS,使得对于SSS中的任意两个点a,ba,ba,b,都存在一个点列{a,v1,v2,…,vk,b}\left\{ a,v_1,v_2,\dots ,v_k,b\right\}{a,v1,v2,…,vk,b}使得这个点列中的每个点都是SSS里面的元素,且序列中相邻两个点间有一条边相连。
在这个前提下,上帝要使得SSS中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过atm的努力,他已经知道了上帝给每棵树上每个节点上的整数。
但是由于atm不擅长计算,他不知道怎样有效地求评分。
他需要你为他写一个程序来计算一棵树的分数。
【输入格式】
第一行一个整数nnn表示这棵树有nnn个节点。
第二行nnn个整数,依次表示每个节点的评分。
接下来n−1n-1n−1行,每行222个整数u,vu,vu,v,表示存在一条uuu到vvv的边。
由于这是一棵树,所以是不存在环的。
树的节点编号从1∼n1\sim n1∼n。
【输出格式】
输出一行一个数,表示上帝给这棵树的分数。
【数据范围】
1≤n≤1051≤n≤10^51≤n≤105
每个节点的评分的绝对值均不超过10610^6106。
【输入样例】
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
【输出样例】
8
【分析】
状态表示:f[u]f[u]f[u]表示以uuu为根的子树中且包含uuu的所有连通块的权值的最大值。
状态计算:假设s1,s2,…,sks_1,s_2,\dots ,s_ks1,s2,…,sk是uuu的子节点,则f[u]=w[u]+max(0,f[s1])+⋯+max(0,f[sk])f[u]=w[u]+max(0,f[s_1])+\dots +max(0,f[s_k])f[u]=w[u]+max(0,f[s1])+⋯+max(0,f[sk])。
最后我们遍历每个节点iii,求出最大的f[i]f[i]f[i]即为答案。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010, M = N << 1;
int e[M], ne[M], h[N], idx;
LL w[N], f[N];
int n;
void add(int u, int v)
{
e[idx] = v, ne[idx] = h[u], h[u] = idx++;
}
void dfs(int u, int fa)
{
f[u] = w[u];
for (int i = h[u]; ~i; i = ne[i])
if (e[i] != fa)
{
dfs(e[i], u);
f[u] += max(0ll, f[e[i]]);
}
}
int main()
{
cin >> n;
memset(h, -1, sizeof h);
for (int i = 1; i <= n; i++) cin >> w[i];
for (int i = 0; i < n - 1; i++)
{
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
dfs(1, -1);
LL res = f[1];
for (int i = 2; i <= n; i++) res = max(res, f[i]);
cout << res << endl;
return 0;
}
五、AcWing 1303. 斐波那契前n项和
【题目描述】
大家都知道Fibonacci数列吧,f1=1,f2=1,f3=2,f4=3,…,fn=fn−1+fn−2f_1=1,f_2=1,f_3=2,f_4=3,\dots ,f_n=f_{n-1}+f_{n-2}f1=1,f2=1,f3=2,f4=3,…,fn=fn−1+fn−2。
现在问题很简单,输入nnn和mmm,求fnf_nfn的前nnn项和Sn mod mS_n\ mod\ mSn mod m。
【输入格式】
共一行,包含两个整数nnn和mmm。
【输出格式】
输出前nnn项和Sn mod mS_n\ mod\ mSn mod m的值。
【数据范围】
1≤n≤2×1091≤n≤2\times 10^91≤n≤2×109
1≤m≤109+101≤m≤10^9+101≤m≤109+10
【输入样例】
5 1000
【输出样例】
12
【分析】
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 3;
int F[N] = { 1, 1, 1 };
int A[N][N] = { { 0, 1, 0 }, { 1, 1, 1 }, { 0, 0, 1 } };
int n, m;
void mul(int c[], int a[], int b[][N])
{
int temp[N] = { 0 };
for (int i = 0; i < N; i++)
for (int j = 0; j < N; j++)
temp[i] = (temp[i] + (LL)a[j] * b[j][i]) % m;
memcpy(c, temp, sizeof temp);
}
void mul(int c[][N], int a[][N], int b[][N])
{
int temp[N][N] = { 0 };
for (int i = 0; i < N; i++)
for (int j = 0; j < N; j++)
for (int k = 0; k < N; k++)
temp[i][j] = (temp[i][j] + (LL)a[i][k] * a[k][j]) % m;
memcpy(c, temp, sizeof temp);
}
int main()
{
cin >> n >> m;
n--;
while (n)
{
if (n & 1) mul(F, F, A);//F = F * A
mul(A, A, A);//A = A * A
n >>= 1;
}
cout << F[2] << endl;
return 0;
}
六、AcWing 1226. 包子凑数
【题目描述】
小明几乎每天早晨都会在一家包子铺吃早餐。
他发现这家包子铺有NNN种蒸笼,其中第iii种蒸笼恰好能放AiA_iAi个包子。
每种蒸笼都有非常多笼,可以认为是无限笼。
每当有顾客想买XXX个包子,卖包子的大叔就会迅速选出若干笼包子来,使得这若干笼中恰好一共有XXX个包子。
比如一共有333种蒸笼,分别能放3,4,53,4,53,4,5个包子。
当顾客想买111111个包子时,大叔就会选222笼333个的再加111笼555个的(也可能选出111笼333个的再加222笼444个的)。
当然有时包子大叔无论如何也凑不出顾客想买的数量。
比如一共有333种蒸笼,分别能放4,5,64,5,64,5,6个包子。
而顾客想买777个包子时,大叔就凑不出来了。
小明想知道一共有多少种数目是包子大叔凑不出来的。
【输入格式】
第一行包含一个整数NNN。
接下来NNN行,每行包含一个整数AiA_iAi。
【输出格式】
输出一个整数代表答案。
如果凑不出的数目有无限多个,输出INF
。
【数据范围】
1≤N≤1001≤N≤1001≤N≤100
1≤Ai≤1001≤A_i≤1001≤Ai≤100
【输入样例1】
2
4
5
【输出样例1】
6
【输入样例2】
2
4
6
【输出样例2】
INF
【样例解释】
对于样例111,凑不出的数目包括:1,2,3,6,7,111,2,3,6,7,111,2,3,6,7,11。
对于样例222,所有奇数都凑不出来,所以有无限多个。
【分析】
题目一看,是个组合问题,是完全背包问题的变形:有若干种物品,每种物品无限个,选的时候每种物品可以选任意个,求是否能够凑出某个重量。
如果有无限个数凑不出来,说明这些数的最大公约数不是111。这里用到裴蜀定理:任意两个数的组合必定是他们的最大公约数的倍数。同样可以推广到更多数:假设这些数的gcdgcdgcd是ggg,那么他们的组合就一定是ggg的倍数,如果g≠1g\ne1g=1,那么必然有无限个数无法被组合出来。
那么gcd=1gcd=1gcd=1呢?最大不能表示出来的数必定有个上界,因为两个数a,ba,ba,b(当gcd(a,b)=1gcd(a,b)=1gcd(a,b)=1时),最大的不能表示出来的数是(a−1)(b−1)−1(a-1)(b-1)-1(a−1)(b−1)−1。当数字更多的时候,这个上界必然更小(因为可选的数字变多了),而999999和989898是100100100内最大的互质的数,(99−1)(98−1)−1<10000(99-1)(98-1)-1<10000(99−1)(98−1)−1<10000,因此这个上界可以选择100001000010000。
那么下面的事情就是看1∼100001\sim 100001∼10000中有多少个数不能被组合出来,回到了刚开始分析的完全背包问题:
状态定义:f[i][j]f[i][j]f[i][j]表示只选前iii个物品是否能凑出重量jjj。
状态计算:
- f[i][j]=f[i−1][j] ∣ f[i−1][j−vi] ∣ f[i−1][j−2vi] ∣ … ∣ f[i−1][j−kvi]f[i][j]=f[i-1][j]\ |\ f[i-1][j-v_i]\ |\ f[i-1][j-2v_i]\ |\ \dots \ |\ f[i-1][j-kv_i]f[i][j]=f[i−1][j] ∣ f[i−1][j−vi] ∣ f[i−1][j−2vi] ∣ … ∣ f[i−1][j−kvi]
- f[i][j−vi]=f[i−1][j−vi] ∣ f[i−1][j−2vi] ∣ … ∣ f[i−1][j−kvi]f[i][j-v_i]=f[i-1][j-v_i]\ |\ f[i-1][j-2v_i]\ |\ \dots \ |\ f[i-1][j-kv_i]f[i][j−vi]=f[i−1][j−vi] ∣ f[i−1][j−2vi] ∣ … ∣ f[i−1][j−kvi]
因此状态转移方程为:f[i][j]=f[i−1][j] ∣ f[i][j−vi]f[i][j]=f[i-1][j]\ |\ f[i][j-v_i]f[i][j]=f[i−1][j] ∣ f[i][j−vi]。
初始化f[0][0]=truef[0][0]=truef[0][0]=true,最后我们遍历f[n][i],(0≤i≤10000)f[n][i],(0\le i\le 10000)f[n][i],(0≤i≤10000),统计出f[n][i]=falsef[n][i]=falsef[n][i]=false的数量即为答案。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 10010;
bool f[N];
int n, g;
int gcd(int a, int b)
{
return b ? gcd(b, a % b) : a;
}
int main()
{
cin >> n;
f[0] = true;
for (int i = 0; i < n; i++)
{
int v;
cin >> v;
g = gcd(g, v);
for (int j = v; j <= 10000; j++) f[j] |= f[j - v];
}
if (g > 1) { puts("INF"); return 0; }
int res = 0;
for (int i = 0; i <= 10000; i++)
if (!f[i]) res++;
cout << res << endl;
return 0;
}
七、AcWing 1070. 括号配对
【题目描述】
Hecy又接了个新任务:BE处理。
BE中有一类被称为GBE。
以下是GBE的定义:
- 空表达式是GBE;
- 如果表达式
A
是GBE,则[A]
与(A)
都是GBE; - 如果
A
与B
都是GBE,那么AB
是GBE。
下面给出一个BE,求至少添加多少字符能使这个BE成为GBE。
注意:BE是一个仅由( ) [ ]
四种字符中的若干种构成的字符串。
【输入格式】
输入仅一行,为字符串BE。
【输出格式】
输出仅一个整数,表示增加的最少字符数。
【数据范围】
对于所有输入字符串,其长度小于100100100。
【输入样例】
[])
【输出样例】
1
【分析】
本题和第三题《密码脱落》基本一样,从当前BE变成GBE需要添加最少字符的数量等价于当前BE变成最长GBE子序列需要去掉字符的数量。即添加的最少字符等价于原字符串长度减去最长GBE子序列的长度。
本题的最长GBE子序列与第三题的最长回文子序列有点区别,本题的序列可以由两个回文序列拼接而成,即不仅([])
属于GBE子序列,[]()
也属于GBE子序列,因此状态计算有一点区别。
状态表示:f[i][j]f[i][j]f[i][j]表示所有在区间[i,j][i,j][i,j]中的GBE子序列的最大长度。
状态计算:
- 当s[i]s[i]s[i]和s[j]s[j]s[j]匹配时,f[i][j]=f[i+1][j−1]+2f[i][j]=f[i+1][j-1]+2f[i][j]=f[i+1][j−1]+2;
- 当s[i]s[i]s[i]和s[j]s[j]s[j]不匹配时,我们枚举两个GBE子序列的拼接点k(i≤k<j)k(i\le k<j)k(i≤k<j),则f[i][j]=f[i][k]+f[k+1][j]f[i][j]=f[i][k]+f[k+1][j]f[i][j]=f[i][k]+f[k+1][j]。在枚举kkk的时候我们也会将状态f[i+1][j]f[i+1][j]f[i+1][j]和f[i][j−1]f[i][j-1]f[i][j−1]进行计算。
初始化:当区间长度为111时,括号不可能匹配,因此f[i][i]=0f[i][i]=0f[i][i]=0,直接从区间长度为222开始计算即可。最后的答案为n−f[0][n−1]n-f[0][n-1]n−f[0][n−1],其中nnn为原字符串的长度。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110;
char s[N];
int f[N][N];
bool check(int l, int r)
{
if (s[l] == '(' && s[r] == ')') return true;
if (s[l] == '[' && s[r] == ']') return true;
return false;
}
int main()
{
cin >> s;
int n = strlen(s);
for (int len = 2; len <= n; len++)//len为1时一定不匹配
for (int i = 0; i + len - 1 < n; i++)
{
int j = i + len - 1;
if (check(i, j)) f[i][j] = f[i + 1][j - 1] + 2;
for (int k = i; k < j; k++)
f[i][j] = max(f[i][j], f[i][k] + f[k + 1][j]);
}
cout << n - f[0][n - 1] << endl;
return 0;
}
八、AcWing 1078. 旅游规划
【题目描述】
W市的交通规划出现了重大问题,市政府下定决心在全市各大交通路口安排疏导员来疏导密集的车流。
但由于人员不足,W市市长决定只在最需要安排人员的路口安排人员。
具体来说,W市的交通网络十分简单,由nnn个交叉路口和n−1n-1n−1条街道构成,交叉路口路口编号依次为0,1,…,n−10,1,\dots ,n-10,1,…,n−1。
任意一条街道连接两个交叉路口,且任意两个交叉路口间都存在一条路径互相连接。
经过长期调查的结果显示,如果一个交叉路口位于W市交通网最长路径上,那么这个路口必定拥挤不堪。
所谓最长路径,定义为某条路径p=(v1,v2,…,vk)p=(v_1,v_2,\dots ,v_k)p=(v1,v2,…,vk),路径经过的路口各不相同,且城市中不存在长度大于kkk的路径(因此最长路径可能不唯一)。
因此W市市长想知道哪些路口位于城市交通网的最长路径上。
【输入格式】
第一行包含一个整数nnn。
之后n−1n-1n−1行每行两个整数u,vu,vu,v,表示编号为uuu和vvv的路口间存在着一条街道。
【输出格式】
输出包括若干行,每行包括一个整数——某个位于最长路径上的路口编号。
为了确保解唯一,请将所有最长路径上的路口编号按编号顺序由小到大依次输出。
【数据范围】
1≤n≤2×1051≤n≤2×10^51≤n≤2×105
【输入样例】
10
0 1
0 2
0 4
0 6
0 7
1 3
2 5
4 8
6 9
【输出样例】
0
1
2
3
4
5
6
8
9
【分析】
变量定义与解释:我们使用dis1[i]dis1[i]dis1[i]表示点iii往下走所能走的最远距离,dis2[i]dis2[i]dis2[i]表示点iii往下走所能走的次远距离(非严格),next1[i]next1[i]next1[i]表示点iii往下走的最远路径上的下一个点的编号,up[i]up[i]up[i]表示点iii往上走所能走的最远距离。
树的直径:即树中的最长路径。求解树的直径的其中一种方式是任选一点xxx,求出距离xxx最远的点yyy,然后再以yyy为起点求出距离yyy最远的点的距离,则该距离就为树的直径,但是这种解法不适用于本题,本题需要求解出可能在树的直径(可能有多条直径)上的所有点。那么我们换个思路,树的直径maxdmaxdmaxd就是所有节点往下走的最大距离与次大距离之和中的最大值,即maxd=max(dis1[i]+dis2[i])maxd=max(dis1[i]+dis2[i])maxd=max(dis1[i]+dis2[i])。
变量求解:求解disdisdis时需要用子节点jjj更新父节点uuu,即uuu往jjj这跳路径上走的最远距离就为dis1[j]+1dis1[j]+1dis1[j]+1,如果该值大于dis1[u]dis1[u]dis1[u],那么dis2[u]=dis1[u],dis1[u]=dis1[j]+1dis2[u]=dis1[u],dis1[u]=dis1[j]+1dis2[u]=dis1[u],dis1[u]=dis1[j]+1,即次远距离先更新为最远距离然后再更新最远距离,然后记录next1[u]=jnext1[u]=jnext1[u]=j;否则如果大于dis2[u]dis2[u]dis2[u]则单独更新dis2dis2dis2即可。求解upupup时需要用父节点uuu更新子节点jjj,jjj往上走有两种情况,一种是走到uuu然后再往上走,即up[j]=up[u]+1up[j]=up[u]+1up[j]=up[u]+1,一种是走到uuu然后往下走,往下走的话肯定是优先走最远的那条路,但是如果最远的那条路就是u→ju→ju→j这条则不能走,因此如果next1[u]≠jnext1[u]\ne jnext1[u]=j,则up[j]=dis1[u]+1up[j]=dis1[u]+1up[j]=dis1[u]+1,否则up[j]=dis2[u]+1up[j]=dis2[u]+1up[j]=dis2[u]+1。
如何判断某个点iii是否在直径上呢?这个点往下走有多条路径,往上走只有一条路径,也就是走到父节点,那么经过这个点的所有路径中最长的那条的长度一定是dis1[i],dis2[i],up[i]dis1[i],dis2[i],up[i]dis1[i],dis2[i],up[i]中较大的两者之和(因为不确定dis2[i]dis2[i]dis2[i]和up[i]up[i]up[i]谁更大因此两者都需要考虑)。如果较大的两个数之和等于树的直径,那么这个点就在某条直径上。
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200010, M = N << 1;
int e[M], ne[M], h[N], idx;
int dis1[N], dis2[N], next1[N], up[N];
int n, maxd;//maxd为直径
void add(int u, int v)
{
e[idx] = v, ne[idx] = h[u], h[u] = idx++;
}
void dfs_dis(int u, int fa)
{
for (int i = h[u]; ~i; i = ne[i])
{
int j = e[i];
if (j != fa)
{
dfs_dis(j, u);
int dis = dis1[j] + 1;
if (dis > dis1[u]) dis2[u] = dis1[u], dis1[u] = dis, next1[u] = j;
else if (dis > dis2[u]) dis2[u] = dis;
}
}
maxd = max(maxd, dis1[u] + dis2[u]);
}
void dfs_up(int u, int fa)
{
for (int i = h[u]; ~i; i = ne[i])
{
int j = e[i];
if (j != fa)
{
up[j] = up[u] + 1;//从u往上走
if (next1[u] != j) up[j] = max(up[j], dis1[u] + 1);//从u往下走最长路
else up[j] = max(up[j], dis2[u] + 1);//从u往下走次长路
dfs_up(j, u);
}
}
}
int main()
{
cin >> n;
memset(h, -1, sizeof h);
for (int i = 0; i < n - 1; i++)
{
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
dfs_dis(0, -1);
dfs_up(0, -1);
for (int i = 0; i < n; i++)
{
int dis[3] = { dis1[i], dis2[i], up[i] };
sort(dis, dis + 3);//排序方便找出三者中最大的两个
if (dis[1] + dis[2] == maxd) cout << i << endl;
}
return 0;
}
九、AcWing 1217. 垒骰子
【题目描述】
赌圣atm晚年迷恋上了垒骰子,就是把骰子一个垒在另一个上边,不能歪歪扭扭,要垒成方柱体。
经过长期观察,atm发现了稳定骰子的奥秘:有些数字的面贴着会互相排斥!
我们先来规范一下骰子:111的对面是444,222的对面是555,333的对面是666。
假设有mmm组互斥现象,每组中的那两个数字的面紧贴在一起,骰子就不能稳定的垒起来。
atm想计算一下有多少种不同的可能的垒骰子方式。
两种垒骰子方式相同,当且仅当这两种方式中对应高度的骰子的对应数字的朝向都相同。
由于方案数可能过多,请输出模109+710^9+7109+7的结果。
【输入格式】
第一行包含两个整数n,mn,mn,m,分别表示骰子的数目和排斥的组数。
接下来mmm行,每行两个整数a,ba,ba,b,表示数字aaa和bbb不能紧贴在一起。
【输出格式】
共一个数,表示答案模109+710^9+7109+7的结果。
【数据范围】
1≤n≤1091≤n≤10^91≤n≤109
1≤m≤361≤m≤361≤m≤36
1≤a,b≤61≤a,b≤61≤a,b≤6
【输入样例】
2 1
1 2
【输出样例】
544
【分析】
【代码】
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 6, MOD = 1e9 + 7;
int A[N][N], F[N];
int op[6] = { 3, 4, 5, 0, 1, 2 };//op[i]表示i对面的数字
int n, m;
void mul(int c[], int a[], int b[][N])
{
int temp[N] = { 0 };
for (int i = 0; i < N; i++)
for (int j = 0; j < N; j++)
temp[i] = (temp[i] + (LL)a[j] * b[j][i]) % MOD;
memcpy(c, temp, sizeof temp);
}
void mul(int c[][N], int a[][N], int b[][N])
{
int temp[N][N] = { 0 };
for (int i = 0; i < N; i++)
for (int j = 0; j < N; j++)
for (int k = 0; k < N; k++)
temp[i][j] = (temp[i][j] + (LL)a[i][k] * b[k][j]) % MOD;
memcpy(c, temp, sizeof temp);
}
int main()
{
cin >> n >> m;
n--;
fill(A[0], A[0] + N * N, 4);
fill(F, F + N, 4);
while (m--)
{
int a, b;
cin >> a >> b;
A[a - 1][op[b - 1]] = 0, A[b - 1][op[a - 1]] = 0;
}
while (n)
{
if (n & 1) mul(F, F, A);
mul(A, A, A);
n >>= 1;
}
LL res = 0;
for (int i = 0; i < N; i++) res = (res + F[i]) % MOD;
cout << res << endl;
return 0;
}