| # |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
| 788541 |
2023-07-20T10:38:19 Z |
79brue |
Paths (RMI21_paths) |
C++17 |
|
418 ms |
53856 KB |
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int L;
int cnt[262200]; ll sum[262200];
ll numbers[270002];
void init(int _L){
L = _L;
}
void update(int x, int v1, ll v2){
while(x<=L){
cnt[x] += v1, sum[x] += v2;
x+=x&-x;
}
}
ll query(int x){
int l = 0;
ll ret = 0;
for(int d=17; d>=0; d--){
int v = cnt[l+(1<<(d+1))] - cnt[l+(1<<d)];
if(v > x) l += (1<<d);
else ret += (sum[l+(1<<(d+1))] - sum[l+(1<<d)]), x -= v;
}
if(x && cnt[l+1]) ret += numbers[l] * x;
return ret;
}
struct Edge{
int s, e; ll v; int idx;
Edge(){}
Edge(int s, int e, ll v): s(s), e(e), v(v){}
bool operator<(const Edge &r)const{
return idx<r.idx;
}
};
int n, k;
Edge arr[200002];
vector<Edge> link[100002];
int in[100002], out[100002], idx[100002], inCnt;
int par[100002];
void dfs_in(int x, int p=-1){
in[x] = ++inCnt;
idx[inCnt] = x;
par[x] = p;
for(auto y: link[x]){
if(y.e == p) continue;
dfs_in(y.e, x);
}
out[x] = inCnt;
}
vector<Edge> linkSet[100002];
ll MX[200002], where[200002]; /// 이 간선 방향으로 갔을 때 최대가 몇인가
vector<pair<ll, ll> > options[100002]; /// 이 정점에서 나갈 수 있는 모든 옵션
pair<ll, int> dfs_getValues(int x, int p=-1){
// printf("Get value %d %d\n", x, p);
if(!linkSet[x].empty()){
vector<Edge> tlst;
for(Edge y: linkSet[x]){
if((y.idx ^ p) == 1){
tlst.push_back(y);
continue;
}
options[x].push_back(make_pair(MX[y.idx] = (dfs_getValues(y.e, y.idx).first + y.v), y.idx));
}
linkSet[x].swap(tlst);
sort(options[x].rbegin(), options[x].rend());
}
if(options[x].empty()) return make_pair(0, -1);
else if((options[x][0].second ^ p) != 1) return options[x][0];
else if((int)options[x].size() == 1) return make_pair(0, -1);
else return options[x][1];
}
vector<ll> inQuery[100002], outQuery[100002];
ll ans[100002];
void putQuery(int s, int e, ll p, int mode){
if(in[s] < in[e]){
if(mode == 1){
inQuery[1].push_back(p), outQuery[in[e]].push_back(p);
if(out[e]+1 <= n+1) inQuery[out[e]+1].push_back(p), outQuery[n+1].push_back(p);
}
else{
outQuery[1].push_back(p), inQuery[in[e]].push_back(p);
if(out[e]+1 <= n+1) outQuery[out[e]+1].push_back(p), inQuery[n+1].push_back(p);
}
}
else{
if(mode == 1) inQuery[in[s]].push_back(p), outQuery[out[s]+1].push_back(p);
else outQuery[in[s]].push_back(p), inQuery[out[s]+1].push_back(p);
}
}
int main(){
scanf("%d %d", &n, &k);
for(int i=1; i<n; i++){
scanf("%d %d %lld", &arr[i*2-2].s, &arr[i*2-2].e, &arr[i*2-2].v);
arr[i*2-1].s = arr[i*2-2].e, arr[i*2-1].e = arr[i*2-2].s, arr[i*2-1].v = arr[i*2-2].v;
arr[i*2-2].idx = i*2-2, arr[i*2-1].idx = i*2-1;
link[arr[i*2-2].s].push_back(arr[i*2-2]);
link[arr[i*2-1].s].push_back(arr[i*2-1]);
}
for(int i=1; i<=n; i++) linkSet[i] = link[i];
dfs_in(1);
for(int i=0; i<(n-1)*2; i++){
if(MX[i]) continue;
pair<ll, int> p = dfs_getValues(arr[i].e, i);
MX[i] = p.first + arr[i].v;
where[i] = p.second;
}
for(int i=1; i<=n; i++) options[i].clear();
for(int i=0; i<(n-1)*2; i++){
options[arr[i].s].push_back(make_pair(MX[i], i));
}
for(int i=1; i<=n; i++) sort(options[i].rbegin(), options[i].rend());
for(int i=1; i<=n; i++){
/// 수는 최대 2N개
for(int j=0; j<(int)options[i].size(); j++){
int p = options[i][j].second; ll v = options[i][j].first;
int s = arr[p].s, e = arr[p].e;
putQuery(s, e, v, 1);
if(where[p] != -1 && v!=arr[p].v) putQuery(s, e, v-arr[p].v, -1);
}
}
numbers[0] = 0;
for(int i=0; i<(n-1)*2; i++) numbers[i+1] = MX[i];
sort(numbers, numbers+n*2-1);
int L = unique(numbers, numbers+n*2-1) - numbers;
for(int i=1; i<=n; i++) for(ll &p: inQuery[i]) p = lower_bound(numbers, numbers+L, p) - numbers;
for(int i=1; i<=n; i++) for(ll &p: outQuery[i]) p = lower_bound(numbers, numbers+L, p) - numbers;
init(1<<18);
for(int i=1; i<=n; i++){
for(ll p: inQuery[i]){
// printf("In update %d %lld\n", i, numbers[p]);
update(p+1, 1, numbers[p]);
}
for(ll p: outQuery[i]){
// printf("Out update %d %lld\n", i, numbers[p]);
update(p+1, -1, -numbers[p]);
}
ans[idx[i]] = query(k);
}
for(int i=1; i<=n; i++){
printf("%lld\n", ans[i]);
}
}
Compilation message
Main.cpp: In function 'int main()':
Main.cpp:105:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
105 | scanf("%d %d", &n, &k);
| ~~~~~^~~~~~~~~~~~~~~~~
Main.cpp:107:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
107 | scanf("%d %d %lld", &arr[i*2-2].s, &arr[i*2-2].e, &arr[i*2-2].v);
| ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
8 ms |
12088 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
8 ms |
12088 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
8 ms |
12088 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
8 ms |
12088 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
418 ms |
53856 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
8 ms |
12088 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
