QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#528873 | #4839. Smaller LCA | Nanani_fan# | WA | 1985ms | 541252kb | C++20 | 3.4kb | 2024-08-23 23:58:22 | 2024-08-23 23:58:23 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
//注意Sum和cnt的区别
const int maxn = 3e5+10; // 数据范围
ll ans[maxn];
int n, cnt;
int sum[maxn*150], ls[maxn*150], rs[maxn*150];
// 用法:update(root, 1, n, x, f); 其中 x 为待修改节点的编号
void update(int& p, int L, int R, int x, int f) { // 引用传参
if (!p) p = ++cnt; // 当结点为空时,创建一个新的结点
if (L == R) {
sum[p] += f;
return;
}
int m = L + ((R - L) >> 1);
if (x <= m)
update(ls[p], L, m, x, f);
else
update(rs[p], m + 1, R, x, f);
sum[p] = sum[ls[p]] + sum[rs[p]]; // pushup
}
// 用法:query(root, 1, n, l, r);
int query(int p, int L, int R, int l, int r) {
if (!p) return 0; // 如果结点为空,返回 0
if (L >= l && R <= r) return sum[p];
int m = L + ((R - L) >> 1), ans = 0;
if (l <= m) ans += query(ls[p], L, m, l, r);
if (r > m) ans += query(rs[p], m + 1, R, l, r);
return ans;
}
int merge(int a, int b, int l, int r) {
if (!a) return b;
if (!b) return a;
if (l == r) {
sum[a]+=sum[b];
return a;
}
int mid = (l + r) >> 1;
ls[a] = merge(ls[a], ls[b], l, mid);
rs[a] = merge(rs[a], rs[b], mid + 1, r);
sum[a]=sum[ls[a]]+sum[rs[a]];
return a;
}
vector<int> ve[maxn];
int L[maxn],R[maxn],dfn;
int Sum[maxn];
void add(int l,int r,int x){
// cout<<"l="<<l<<" r="<<r<<" x="<<x<<endl;
Sum[l]+=x;
Sum[r+1]-=x;
}
int dep[maxn],f[21][maxn];
void dfs1(int x,int h)
{
L[x]=++dfn;
dep[x]=dep[h]+1;
for(int i=0;i<=19;i++)
f[i+1][x]=f[i][f[i][x]];
for(auto it:ve[x])if(it!=h){
f[0][it]=x;
dfs1(it,x);
}
R[x]=dfn;
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=20;i>=0;i--)
{
if(dep[f[i][x]]>=dep[y]) x=f[i][x];
if(x==y) return x;
}
for(int i=20;i>=0;i--)
if(f[i][x]!=f[i][y])
x=f[i][x],y=f[i][y];
return f[0][x];
}
int getson(int x,int y){
assert(L[y]>=L[x]&&R[y]<=R[x]);
for(int i=20;i>=0;i--)if(dep[f[i][y]]>dep[x])y=f[i][y];
return y;
}
vector<int> ad[maxn];
vector<pair<int,int>>era[maxn];
int dfs(int x,int h){
int rt=0;
for(auto it:ve[x])if(it!=h){
int g=dfs(it,x);
int t=query(g,1,n,x,n);
add(L[x],R[x],t);
add(L[it],R[it],-t);
rt=merge(rt,g,1,n);
}
for(auto [p,q]:era[x])if(x<=p*q){
add(1,n,1);add(L[x],R[x],-1);
if(q==x)swap(p,q);
update(rt,1,n,p*q,-2);
if(p==x)add(L[x],R[x],-1);
}
for(auto it:ad[x]){
update(rt,1,n,it,1);
add(L[x],R[x],1);
}
return rt;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n;
for(int i=1;i<n;i++){
int x,y;cin>>x>>y;
// int x=i+1,y=rand()%(i)+1;
ve[x].push_back(y);
ve[y].push_back(x);
}
ll all=0;
dfs1(1,0);
for(int i=1;i<=n;i++){
all++;
for(int j=i+1;j<=n;j++)if((ll)i*j<=n){
ad[i].push_back(i*j);
ad[j].push_back(i*j);
int l=lca(i,j);
era[l].emplace_back(i,j);
}
else {all+=n-j+1;break;}
}
dfs(1,0);
for(int i=1;i<=n;i++)Sum[i]+=Sum[i-1];
for(int i=1;i<=n;i++){
cout<<all+Sum[L[i]]<<"\n";
// cout<<all<<" "<<Sum[L[i]]<<endl;
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 3ms
memory: 32416kb
input:
5 1 2 4 2 2 5 3 5
output:
15 15 15 15 14
result:
ok 5 number(s): "15 15 15 15 14"
Test #2:
score: -100
Wrong Answer
time: 1985ms
memory: 541252kb
input:
300000 40632 143306 32259 40632 225153 143306 269774 225153 289668 40632 191636 269774 85717 191636 58564 191636 156509 143306 289939 40632 247103 269774 40257 40632 98149 289668 142277 143306 291616 40257 46813 225153 56324 143306 277154 142277 53903 289668 114266 32259 152231 58564 241151 152231 4...
output:
45000862900 45000570621 45000616394 45000615413 44999659757 45000475013 45000068233 44999748392 44999633097 44999787635 45000596851 44999795183 44999930046 44999590901 45000369394 45000123880 44999778255 44999975906 44999986907 44999782119 45000513522 44999971580 45000117181 44999745238 44999939366 ...
result:
wrong answer 1st numbers differ - expected: '44999437117', found: '45000862900'