codeforces 894D Ralph And His Tour in Binary Country
D. Ralph And His Tour in Binary Country
time limit per test
2.5 seconds
memory limit per test
512 megabytes
input
standard input
output
standard output
Ralph is in the Binary Country. The Binary Country consists of n cities and (n - 1) bidirectional roads connecting the cities. The roads are numbered from 1 to (n
Now Ralph gives you m queries. In each query he tells you some city A
Ralph is interested in tours from Ai in which he can gain positive happiness. For each query, compute the sum of happiness gains for all such tours.
Ralph will never take the same tour twice or more (in one query), he will never pass the same city twice or more in one tour.
Input
The first line contains two integers n and m (1 ≤ n ≤ 106, 1 ≤ m ≤ 105).
(n - 1) lines follow, each line contains one integer Li (1 ≤ Li ≤ 105), which denotes the length of the i-th road.
m lines follow, each line contains two integers Ai and Hi (1 ≤ Ai ≤ n, 0 ≤ Hi ≤ 107).
Output
Print m lines, on the i-th line print one integer — the answer for the i-th query.
Examples
input
Copy
2 2 5 1 8 2 4
output
Copy
11 4
input
Copy
6 4 2 1 1 3 2 2 4 1 3 3 2 1 7
output
Copy
11 6 3 28
Note
Here is the explanation for the second sample.
Ralph's first query is to start tours from city 2 and Hi equals to 4. Here are the options:
- He can choose city 5 as his terminal city. Since the distance between city 5 and city 2 is 3, he can gain happiness 4 - 3 = 1.
- He can choose city 4 as his terminal city and gain happiness 3.
- He can choose city 1 as his terminal city and gain happiness 2.
- He can choose city 3 as his terminal city and gain happiness 1.
- Note that Ralph can choose city 2 as his terminal city and gain happiness 4.
- Ralph won't choose city 6 as his terminal city because the distance between city 6 and city 2 is 5, which leads to negative happiness for Ralph.
So the answer for the first query is 1 + 3 + 2 + 1 + 4 = 11.
題目大意:
#include <cstdio>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
#define ll long long
const int maxn = 1e6 + 5;
vector<int> L;
vector<int> dist[maxn];
vector<ll> ps[maxn];
int A, H;
int n, m;
ll bs(int i, int x) {
if (x <= 0) return 0;
int index = upper_bound(dist[i].begin(), dist[i].end(), x) - dist[i].begin();
return 1ll * x * index - ps[i][index - 1];
}
int main() {
scanf("%d%d", &n, &m);
int v;
L.push_back(0);
L.push_back(0);
for (int i = 1; i < n; i++) {
scanf("%d", &v);
L.push_back(v);
}
for (int i = n; i > 0; i--) {
int lson = i * 2;
int rson = i * 2 + 1;
dist[i].push_back(0);
if (lson <= n) {
for (int j = 0; j < dist[lson].size(); j++) {
dist[i].push_back(dist[lson][j] + L[lson]);
}
}
if (rson <= n) {
for (int j = 0; j < dist[rson].size(); j++) {
dist[i].push_back(dist[rson][j] + L[rson]);
}
}
sort(dist[i].begin(), dist[i].end());
// ll tmp = 0;
// for (int j = 0; j < dist[i].size(); j++) {
// ps[i].push_back(dist[i][j] + tmp);
// tmp = ps[i][j];
// }
ps[i].resize(dist[i].size());
for(int j = 1; j < ps[i].size(); ++j) {
ps[i][j] = ps[i][j - 1] + dist[i][j];
}
}
// for (int i = 1; i <= n; i++)
// {
// for (int j = 0; j < ps[i].size(); j++)
// cout << ps[i][j] << " ";
// cout << endl;
// }
while (m--) {
scanf("%d%d", &A, &H);
ll ans = 0;
int last = 0;
while (A > 0) {
if (H < 0) break;
ans += H;
int lson = A * 2;
int rson= A * 2 + 1;
if (lson <= n && lson != last) {
ans += bs(lson, H - L[lson]);
// cout << "debug " << lson << " " << H - L[lson] << " " << ans << endl;
}
if (rson <= n && rson != last) {
ans += bs(rson, H - L[rson]);
// cout << "debug " << rson << " " << H - L[rson] << " " << ans << endl;
}
H -= L[A];
last = A;
A /= 2;
}
printf("%I64d\n", ans);
}
return 0;
}