题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805388447170560
树的DFS,应该不算难。但由于本人的陋习,这种涉及叶子到根的DFS都喜欢从下往上搜(即从叶子搜索到根)(感觉是为了节省空间),最后一个点超时了。看了柳神的解法也是要至少双层数组,强迫症犯了,没有照抄,把自己的自底向上DFS剪了减枝,然后就过了(耶~)
读入数据,parent[]
数组保存每个节点的父亲,lvl[]
保存每个节点的深度(层数)。根的深度为0
int N;
double P, r;
scanf("%d %lf %lf", &N, &P, &r);
vector<int> parent(N, -1);
vector<Retailer> rtl;
vector<int> lvl(N, -1);
lvl[0] = 0;
rtl[]
数组存叶子节点的编号和销售量,也就是retailers
class Retailer {
public:
int id;
int amt;
};
读入数据,如果是叶子节点(K==0
),就放入rtl[]
数组中;如果不是,就记录将它的孩子的parent值置为它,记录每个节点的父亲。
for (int i = 0; i < N; i++) {
int K;
scanf("%d", &K);
if (K == 0) {
Retailer tmp;
scanf("%d", &tmp.amt);
tmp.id = i;
rtl.push_back(tmp);
continue;
}
for (int j = 0; j < K; j++) {
int kid;
scanf("%d", &kid);
parent[kid] = i;
}
}
对每一个叶子节点,找它的父亲节点,若父亲节点的层数已经记录了,就直接用其层数+1计算销售额;否则先获取父亲节点的层数。
double ret = 0.0;
for (int i = 0; i < rtl.size(); i++) {
double sale = rtl[i].amt * P;
int nxt = parent[rtl[i].id];
while (nxt != -1) {
if (lvl[nxt] >= 0) {
sale *= pow((1 + r / 100.0), lvl[nxt] + 1);
break;
}
else
getLvl(nxt, parent, lvl);
}
ret += sale;
}
获取层数函数,如果父亲节点层数已知,就置为父亲层数+1;否则继续向上直到根。(根的层数已初始置为0)
int getLvl(int nxt, vector<int>& parent, vector<int>& lvl) {
if (lvl[parent[nxt]] >= 0)
lvl[nxt] = lvl[parent[nxt]] + 1;
else
lvl[nxt] = getLvl(parent[nxt], parent, lvl) + 1;
return lvl[nxt];
}
因为有了lvl[]
记录层数,从叶子节点出发,就不需要每次都追溯到根了,节省了时间,测试点6就可以过了。
完整代码
#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<map>
#include<set>
#include<queue>
#include<string.h>
using namespace std;
class Retailer {
public:
int id;
int amt;
};
int getLvl(int nxt, vector<int>& parent, vector<int>& lvl) {
if (lvl[parent[nxt]] >= 0)
lvl[nxt] = lvl[parent[nxt]] + 1;
else
lvl[nxt] = getLvl(parent[nxt], parent, lvl) + 1;
return lvl[nxt];
}
int main() {
int N;
double P, r;
scanf("%d %lf %lf", &N, &P, &r);
vector<int> parent(N, -1);
vector<Retailer> rtl;
vector<int> lvl(N, -1);
lvl[0] = 0;
for (int i = 0; i < N; i++) {
int K;
scanf("%d", &K);
if (K == 0) {
Retailer tmp;
scanf("%d", &tmp.amt);
tmp.id = i;
rtl.push_back(tmp);
continue;
}
for (int j = 0; j < K; j++) {
int kid;
scanf("%d", &kid);
parent[kid] = i;
}
}
double ret = 0.0;
for (int i = 0; i < rtl.size(); i++) {
double sale = rtl[i].amt * P;
int nxt = parent[rtl[i].id];
while (nxt != -1) {
if (lvl[nxt] >= 0) {
sale *= pow((1 + r / 100.0), lvl[nxt] + 1);
break;
}
else
getLvl(nxt, parent, lvl);
}
ret += sale;
}
printf("%.1f", ret);
return 0;
}