QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#292332#7118. Closing TimeGoldenglow14270 173ms138200kbC++146.2kb2023-12-28 01:11:232023-12-28 01:11:24

Judging History

你现在查看的是测评时间为 2023-12-28 01:11:24 的历史记录

  • [2024-04-28 08:01:48]
  • 管理员手动重测本题所有提交记录
  • 测评结果:0
  • 用时:186ms
  • 内存:141052kb
  • [2023-12-28 01:11:24]
  • 评测
  • 测评结果:0
  • 用时:173ms
  • 内存:138200kb
  • [2023-12-28 01:11:23]
  • 提交

answer

/*
ID: Victor Chen [mail_vi1]
PROG: Closing Time
LANG: C++
*/

#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
#include <cassert>

#define INF 0x3f3f3f3f

using namespace std;

typedef long long ll;

const int Maxn = 2e5;

class Graph
{
    public:
        int cnt, head[Maxn+10];
        struct Edge
        {
            int to, nxt;
            ll val;
        }p[2*Maxn+10];

        int root;
        int dep[Maxn+10], fa[Maxn+10];
        ll val[Maxn+10];

        void AddEdge(int x, int y, ll v)
        {
            cnt++;
            p[cnt].to = y;
            p[cnt].val = v;

            p[cnt].nxt = head[x];
            head[x] = cnt;
        }

        void dfs(int x)
        {
            for(int i=head[x]; i!=0; i=p[i].nxt)
                if(dep[p[i].to] == 0)
                {
                    dep[p[i].to] = dep[x] + 1;
                    fa[p[i].to] = x;
                    val[p[i].to] = val[x] + p[i].val;
                    dfs(p[i].to);
                }
        }

        void clear()
        {
            cnt = 0;
            memset(head, 0, sizeof(head));
        }

        void init(int x)
        {
            root = x;
            memset(dep, 0, sizeof(dep));
            dep[root] = 1;
            fa[root] = 0;
            val[root] = 0;

            dfs(root);
        }
}g;
class SegTree
{
    public:
        int cnt, root;
        struct Node
        {
            ll l, r;
            int val, ch[2];

            ll sum;
        }p[Maxn*20];

        ll build(ll l, ll r)
        {
            cnt++;
            p[cnt].l = l, p[cnt].r = r;
            p[cnt].val = p[cnt].ch[0] = p[cnt].ch[1] = p[cnt].sum = 0;

            assert(cnt <= Maxn*20);

            return cnt;
        }

        void buildtree(ll l, ll r)
        {
            cnt = 0;
            root = build(l, r);
        }

        void modify(int x, ll k, int v)
        {
            if(p[x].l == p[x].r)
            {
                p[x].val += v;
                p[x].sum += p[x].l * v;
                return;
            }

            ll mid = (p[x].l+p[x].r)/2;
            if(k <= mid)
            {
                if(!p[x].ch[0]) p[x].ch[0] = build(p[x].l, mid);
                modify(p[x].ch[0], k, v);
            }
            else
            {
                if(!p[x].ch[1]) p[x].ch[1] = build(mid+1, p[x].r);
                modify(p[x].ch[1], k, v);
            }

            p[x].val = p[p[x].ch[0]].val + p[p[x].ch[1]].val;
            p[x].sum = p[p[x].ch[0]].sum + p[p[x].ch[1]].sum;
        }

        int find_cnt_below_k(int x, ll k)
        {
            if(k >= p[x].sum)
                return p[x].val;
            if(p[x].l == p[x].r)
                return min((int)(k/p[x].l), p[x].val);
            if(x == 0)
                return 0;

            if(p[x].ch[0])
            {
                if(k == p[p[x].ch[0]].sum) return p[p[x].ch[0]].val;
                else if(k < p[p[x].ch[0]].sum) return find_cnt_below_k(p[x].ch[0], k);
                else return p[p[x].ch[0]].val + find_cnt_below_k(p[x].ch[1], k-p[p[x].ch[0]].sum);
            }
            else
                return find_cnt_below_k(p[x].ch[1], k);
        }
}tree;

int n;
int c1, c2;

ll cst;

ll val[Maxn+10];

priority_queue<ll, vector<ll>, greater<ll>> q;

ll dis[Maxn+10];
ll dis1[Maxn+10], dis2[Maxn+10];

struct Node
{
    int id;
    ll cst1, cst2;
    
    bool operator < (const Node &x) const
    {
        if(cst2 != x.cst2)
            return cst2 < x.cst2;
        else
            return cst1 < x.cst1;
    }

    void build(int id, ll x, ll y)
    {
        this->id = id;
        this->cst1 = min(x, y);
        this->cst2 = max(x, y);
    }
}p[Maxn+10];

int inQueue = 0;

bool mk[Maxn+10];

ll cur[Maxn+10];
int max_score(int N, int X, int Y, ll K, vector<int> U, vector<int> V, vector<int> W)
{
    tree.buildtree(0, 2e11);

    int ans1 = 0, ans2 = 0;

    n = N, c1 = X+1, c2 = Y+1;
    for(int i=0; i<n-1; i++)
    {
        g.AddEdge(U[i], V[i], W[i]);
        g.AddEdge(V[i], U[i], W[i]);
    }

    g.init(c1);
    for(int i=1; i<=n; i++)
        dis1[i] = g.val[i];
    g.init(c2);
    for(int i=1; i<=n; i++)
        dis2[i] = g.val[i];
    
    // Case 1: If not connected.
    for(int i=1; i<=n; i++)
        val[i] = min(dis1[i], dis2[i]);
    sort(val+1, val+n+1);
    cst = 0;
    for(int i=1; i<=n; i++)
        if(cst + val[i] <= K)
            cst += val[i], ans1++;
        else
            break;
        
    // Case 2: If connected.
    for(int i=1; i<=n; i++)
        p[i].build(i, dis1[i], dis2[i]);
    
    memset(mk, 0, sizeof(mk));

    for(int i=c1; i!=c2; i=g.fa[i])
        mk[i] = 1;
    mk[c2] = 1;

    inQueue = 0;
    sort(p+1, p+n+1);
    for(int i=1; i<=n; i++)
    {
        if(mk[p[i].id])
        {
            tree.modify(1, p[i].cst2-p[i].cst1, 1);
            K -= p[i].cst1;
            inQueue++;
        }
        else
            tree.modify(1, p[i].cst1, 1);
    }

    if(K < 0) return ans1;

    ans2 = max(ans2, inQueue + tree.find_cnt_below_k(1, K));

    for(int i=1; i<=n; i++)
    {
        if(mk[p[i].id])
            continue;

        tree.modify(1, p[i].cst1, -1);
        tree.modify(1, p[i].cst2-p[i].cst1, 1);

        K -= p[i].cst1;
        inQueue++;

        if(K < 0)
            break;

        ans2 = max(ans2, tree.find_cnt_below_k(1, K) + inQueue);

        // printf("%d: %d\n", K, tree.find_cnt_below_k(1, K));
    }

    return max(ans1, ans2);
}

// int T;
// int N, X, Y;
// ll K;
// vector<int> U, V, W;

// int main()
// {
//     scanf("%d", &T);
//     while(T != 0)
//     {
//         T--;
//         scanf("%d%d%d%lld", &N, &X, &Y, &K);
//         U.clear(); V.clear(); W.clear();
//         for(int i=0; i<N-1; i++)
//         {
//             int x, y, z;
//             scanf("%d%d%d", &x, &y, &z);
//             U.push_back(x); V.push_back(y); W.push_back(z);
//         }

//         printf("%d\n", max_score(N, X, Y, K, U, V, W));
//     }

//     return 0;
// }

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 0
Wrong Answer
time: 173ms
memory: 138200kb

input:

cc61ad56a4797fb3f5c9529f73ce6fcedd85669b
1
200000 31011 61157 8517583098
31011 129396 964383
1655 129396 331139
1655 191487 566483
110385 191487 865248
43212 110385 542661
43212 81682 13766
81682 91774 546589
91774 124706 780638
124706 175650 118706
10421 175650 615314
10421 151953 436270
140430 151...

output:

081ce3c351cbf526b37954b9ad30f2b531a7585c
OK
632

result:

wrong answer 1st lines differ - on the 1st token, expected: '451', found: '632'

Subtask #2:

score: 0
Wrong Answer

Test #4:

score: 0
Wrong Answer
time: 1ms
memory: 4168kb

input:

cc61ad56a4797fb3f5c9529f73ce6fcedd85669b
1
50 23 25 382806473
0 1 375710
1 2 898637
2 3 10402
3 4 536577
4 5 385023
5 6 71075
6 7 543368
7 8 301497
8 9 174394
9 10 711312
10 11 923006
11 12 675532
12 13 838667
13 14 565729
14 15 979816
15 16 862618
16 17 576015
17 18 177751
18 19 306989
19 20 881492...

output:

081ce3c351cbf526b37954b9ad30f2b531a7585c
OK
100

result:

wrong answer 1st lines differ - on the 1st token, expected: '96', found: '100'

Subtask #3:

score: 0
Skipped

Dependency #2:

0%

Subtask #4:

score: 0
Skipped

Dependency #3:

0%

Subtask #5:

score: 0
Wrong Answer

Test #36:

score: 0
Wrong Answer
time: 1ms
memory: 4132kb

input:

cc61ad56a4797fb3f5c9529f73ce6fcedd85669b
1
4 0 1 9
0 2 2
1 2 3
2 3 3

output:

081ce3c351cbf526b37954b9ad30f2b531a7585c
OK
7

result:

wrong answer 1st lines differ - on the 1st token, expected: '6', found: '7'

Subtask #6:

score: 0
Skipped

Dependency #2:

0%

Subtask #7:

score: 0
Skipped

Dependency #3:

0%

Subtask #8:

score: 0
Skipped

Dependency #4:

0%

Subtask #9:

score: 0
Skipped

Dependency #1:

0%