QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#166938#7103. Red Black TreeOMoonStarsWA 802ms29476kbC++143.1kb2023-09-06 21:17:132023-09-06 21:17:13

Judging History

你现在查看的是最新测评结果

  • [2023-09-06 21:17:13]
  • 评测
  • 测评结果:WA
  • 用时:802ms
  • 内存:29476kb
  • [2023-09-06 21:17:13]
  • 提交

answer

#include<algorithm>
#include<array>
#include<bitset>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#include<numeric>
#include<queue>
#include<set>
#include<unordered_map>
#define fi first
#define se second
#define pb push_back
#define endl '\n'
using namespace std;
using ll=long long;
using db=double;
using pii=pair<int,int>;
const int inf=0x3f3f3f3f;
const int mod=998244353;
const int N=1e5+10,M=20;
bool red[N];//红点
vector<int>g[N],w[N];//邻接表,边权
int nr[N];
ll cost[N];//到最近红祖先的距离
int dep[N];//距根节点距离
int fa[N][M];//二维logn
void dfs1(int u,int f)//当前节点和父节点
{
    dep[u]=dep[f]+1;//深度比父节点多1
    fa[u][0]=f;//第2^0=1个祖先就是父节点
    for(int i=1;i<20;i++)//倍增logn次
        fa[u][i]=fa[fa[u][i-1]][i-1];//第2^i个祖先是第2^(i-1)个祖先的第2^(i-1)个祖先
    for(auto v:g[u])//遍历子节点dfs
        if(v!=f)dfs1(v,u);
}
void dfs2(int u,int f,int r,ll d)//当前节点,父节点,当前距离
{
    if(red[u])r=u,d=0;
    nr[u]=r;
    cost[u]=d;
    for(int i=0;i<g[u].size();i++)
        if(g[u][i]!=f)
            dfs2(g[u][i],u,r,d+w[u][i]);
}
int lca(int x,int y)//倍增求x和y的lca节点
{
    if(dep[x]>dep[y])swap(x,y);//令y比x深
    int dist=dep[y]-dep[x];//深度差值
    for(int i=0;dist;i++,dist>>=1)
        if(dist&1)y=fa[y][i];//将深度二进制拆分上跳,令x和y在同一深度
    if(x==y)return x;//已相等说明是祖先
    for(int i=19;x!=y&&~i;i--)
        if(fa[x][i]!=fa[y][i])//找到第一个不是它们祖先的两个点
        {
            x=fa[x][i];//上跳
            y=fa[y][i];
        }
    if(x!=y)x=fa[x][0];//所有次幂祖先均相同则父节点为lca
    return x;//返回最近公共祖先
}
void solve()
{
    int n,m,q;
    cin >> n >> m >> q;
    for(int i=1;i<=n;i++)//初始化
    {
        red[i]=false;
        g[i].clear();
        w[i].clear();
    }
    while(m--)
    {
        int r;
        cin >> r;
        red[r]=true;
    }
    for(int i=1;i<n;i++)
    {
        int u,v,wi;
        cin >> u >> v >> wi;
        g[u].pb(v);
        w[u].pb(wi);
        g[v].pb(u);
        w[v].pb(wi);
    }
    dfs1(1,0);//预处理每点第2的整数次幂个祖先
    dfs2(1,0,1,0);//预处理每个点到最近红祖先的距离
    while(q--)
    {
        int k;
        cin >> k;
        vector<int>v(k);
        for(int i=0;i<k;i++)
        {
            int u;
            cin >> u;
            v.pb(u);
        }
        sort(v.begin(),v.end(),[](int x,int y){
            return cost[x]>cost[y];
        });
        v.pb(0);
        ll ans=cost[v[1]];
        int lc=v[0];
        for(int i=1;i<k;i++)
        {
            lc=lca(lc,v[i]);
            if(dep[lc]<=dep[nr[v[0]]])break;
            ans=min(ans,max(cost[v[0]]-cost[lc],cost[v[i+1]]));
        }
        cout << ans << endl;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int t;
    cin >> t;
    while(t--)
        solve();
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 9136kb

input:

2
12 2 4
1 9
1 2 1
2 3 4
3 4 3
3 5 2
2 6 2
6 7 1
6 8 2
2 9 5
9 10 2
9 11 3
1 12 10
3 3 7 8
4 4 5 7 8
4 7 8 10 11
3 4 5 12
3 2 3
1 2
1 2 1
1 3 1
1 1
2 1 2
3 1 2 3

output:

4
5
3
8
0
0
0

result:

ok 7 lines

Test #2:

score: -100
Wrong Answer
time: 802ms
memory: 29476kb

input:

522
26 1 3
1
1 4 276455
18 6 49344056
18 25 58172365
19 9 12014251
2 1 15079181
17 1 50011746
8 9 2413085
23 24 23767115
22 2 26151339
26 21 50183935
17 14 16892041
9 26 53389093
1 20 62299200
24 18 56114328
11 2 50160143
6 26 14430542
16 7 32574577
3 16 59227555
3 15 8795685
4 12 5801074
5 20 57457...

output:

148616264
148616264
0
319801028
319801028
255904892
317070839
1265145897
1265145897
1072765445
667742619
455103436
285643094
285643094
285643094
317919339
0
785245841
691421476
605409472
479058444
371688030
303203698
493383271
919185207
910180170
919185207
121535083
181713164
181713164
181713164
181...

result:

wrong answer 691st lines differ - expected: '549790714', found: '374667774'