Submission #1308448

#	Time	Username	Problem	Language	Result	Execution time	Memory
1308448		ender_shayan	Designated Cities (JOI19_designated_cities)	C++20	23 / 100	1355 ms	17564 KiB

designated_cities

#include <bits/stdc++.h>

using namespace std;

// #include <ext/pb_ds/assoc_container.hpp>
// #include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;

typedef long long ll;
typedef long double	ld;
typedef pair<int, int>	pii  ;
typedef pair<ll, ll>	pll  ;
typedef vector<pii>     vii  ;
typedef vector<int>     veci ;
typedef vector<pll>     vll  ;
typedef vector<ll>      vecll;

// find_by_order             order_of_key

//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
#define F		        first
#define S		        second
#define pb		        push_back
#define endl            '\n'
#define Mp		        make_pair
#define all(x)          x.begin(), x.end()
#define debug(x)        cerr << #x << " = " << x << endl
#define set_dec(x)	    cout << fixed << setprecision(x);
#define fast_io         ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file_io         freopen("in.txt" , "r" , stdin) ; freopen("out.txt" , "w" , stdout);
#define lb              lower_bound
#define ub              upper_bound
#define for1(n)         for(int i=1;i<=n;i++)
#define for0(n)         for(int i=0;i<n;i++)
#define forn(n)         for(int i=n;i>0;i--)
#define pq              priority_queue <pii, vector<pii>, greater<pii>>


const ll mod = 1e9+7 ;// 998244353 ;// 1e9+9;

ll inf=1e18;
const int N=1<<17,L=21,bs=701,NN=1e6;
int A[N],B[N],C[N],D[N],E[N],n,m,k,q,dp[N],pre[N],dist[N],par[N],parw[N],st[N],en[N],t,pos[N];
vector<pii>g[N];
ll seg[N*2],lz[N*2];
ll ans;
ll anss[N];
void shift(int l,int r,int v){
    if(r-l==1 || lz[v]==0)return;
    int mid=(l+r)>>1;
    seg[v<<1]+=lz[v];
    seg[v<<1|1]+=lz[v];
    lz[v<<1]+=lz[v];
    lz[v<<1|1]+=lz[v];
    lz[v]=0;
}
void build(int l=1,int r=n+1,int v=1){
    seg[v]=0;lz[v]=0;
    if(r-l==1)return;
    int mid=(l+r)>>1;
    build(l,mid,v<<1);
    build(mid,r,v<<1|1);
}
void upd(int lx,int rx,int x,int l=1,int r=n+1,int v=1){
    if(lx>=r || rx<=l)return;
    if(lx<=l && r<=rx){
        seg[v]+=x;
        lz[v]+=x;
        return;
    }
    shift(l,r,v);
    int mid=(l+r)>>1;
    upd(lx,rx,x,l,mid,v<<1);
    upd(lx,rx,x,mid,r,v<<1|1);
    seg[v]=max(seg[v<<1],seg[v<<1|1]);
}
int get(int l=1,int r=n+1,int v=1){
    if(r-l==1)return l;
    shift(l,r,v);
    int mid=(l+r)>>1;
    if(seg[v<<1]>seg[v<<1|1])return get(l,mid,v<<1);
    return get(mid,r,v<<1|1);
}
void dfs(int v){
    st[v]=++t;pos[t]=v;
    for(pii p:g[v])if(p.F!=par[v]){
        int u=p.F;
        par[u]=v;
        parw[u]=p.S;
        ans+=p.S;
        dfs(u);
        upd(st[u],en[u]+1,p.S);
    }
    en[v]=t;
}
int main(){
    fast_io
    cin>>n;k=n;
    for1(n-1){
        int u,v,c,d;cin>>u>>v>>c>>d;
        g[u].pb({v,c});
        g[v].pb({u,d});
    }
    for1(k)anss[i]=inf;
    for(int root=1;root<=n;root++){
        build();t=0;ans=0;
        par[root]=0;
        dfs(root);
        anss[1]=min(anss[1],ans);
        for(int j=2;j<=k;j++){
            int v=get();
            ll w=seg[1];
            ans-=w;
            v=pos[v];
            while(w!=0){
                upd(st[v],en[v]+1,-parw[v]);
                w-=parw[v];
                v=par[v];
            }
            anss[j]=min(anss[j],ans);
        }
    }
    cin>>q;
    while(q--){
        int x;cin>>x;
        cout<<anss[x]<<endl;
    }




}

• Subtask 1: 6 / 6
• Subtask 2: 0 / 7
• Subtask 3: 0 / 9
• Subtask 4: 17 / 17
• Subtask 5: 0 / 17
• Subtask 6: 0 / 44