답안 #715300

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
715300 2023-03-26T11:52:56 Z Ahmed57 Islands (IOI08_islands) C++14
24 / 100
676 ms 131072 KB
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3")

using namespace std;
#define int long long
stack<int> st;
vector<vector<int>> all;
vector<pair<int,pair<long long,long long>>> adj[1000001];
int tmp[1000001];
int vis[1000001];
bool ss = 0;
void cycle(int i,int pr){
    st.push(i);
    if(tmp[i]){
        vector<int> cyc;
        cyc.push_back(i);
        st.pop();
        while(st.top()!=i){
            cyc.push_back(st.top());st.pop();
        }
        for(int j = cyc.size()-1;j>=0;j--)st.push(cyc[j]);
        all.push_back(cyc);
        st.pop();
        ss = 1;
        return;
    }
    tmp[i] = 1;
    for(auto j:adj[i]){
        if(j.second.second==pr)continue;
        cycle(j.first,j.second.second);
        if(ss){
            tmp[i] = 0;
            st.pop();
            return ;
        }
    }
    tmp[i] = 0;
    st.pop();
}
void fil(int i){
    vis[i]=1;
    for(auto j:adj[i]){
        if(!vis[j.first]){
            fil(j.first);
        }
    }
}
long long maxDep[1000005];
long long TreeDim[1000005];
int ele[1000005];
long long sst,maa,node;
void ma(int i,int pr,long long dep){
    maxDep[sst] =max(maxDep[sst],dep);
    for(auto j:adj[i]){
        if(j.first==pr||ele[j.first]==1)continue;
        ma(j.first,i,dep+j.second.first);
    }
}
void Dim(int i,int pr,long long dep){
    if(maa<dep){
        maa =dep;
        node = i;
    }
    TreeDim[sst] =max(TreeDim[sst],dep);
    for(auto j:adj[i]){
        if(j.first==pr||ele[j.first]==1)continue;
        Dim(j.first,i,dep+j.second.first);
    }
}
signed main(){
    int n;cin>>n;
    map<pair<int,int>,long long>mp;
    for(int i = 1;i<=n;i++){
        long long a,b;
        cin>>a>>b;
        mp[{i,a}] = b;
        mp[{a,i}] = b;
        adj[i].push_back({a,{b,i}});
        adj[a].push_back({i,{b,i}});
    }
    for(int i = 1;i<=n;i++){
        if(!vis[i]){
            ss = 0;
            cycle(i,-1);
            fil(i);
        }
    }

    for(auto i:all){
        int len = i.size();
        for(int j = 0;j<i.size();j++){
            ele[i[(j+1)%len]] = 1;
            ele[i[((j-1)%len+len)%len]] = 1;
            sst = i[j];maa = -1,node = i[j];
            ma(i[j],-1,0);
            Dim(node,-1,0);
            maa = -1;
            Dim(node,-1,0);
            ele[i[(j+1)%len]] = 0;
            ele[i[((j-1)%len+len)%len]] = 0;
        }
    }
    long long sum = 0;
    long long cost[1000001];
    for(auto i:all){
        if(i.size()==2){
            long long ans = max(TreeDim[i[0]],TreeDim[i[1]]);
            long long e = 0;
            for(auto j:adj[i[0]]){
                if(j.first==i[1])e = max(e,j.second.first);
            }
            ans = max(ans,maxDep[i[0]]+maxDep[i[1]]+e);
            sum+=ans;
            continue;
        }
        long long ans = 0;
        for(auto j:i)ans = max(ans,TreeDim[j]);
        //cout<<ans<<endl;
        multiset<pair<long long,int>> s;
        long long sz = mp[{i[0],i[1]}];
        int len = i.size();
        for(int j = 1;j<i.size();j++){
            //cout<<i[j]<<"hh"<<(-sz)+maxDep[i[j]]<<endl;
            s.insert({(-sz)+maxDep[i[j]],i[j]});
            cost[i[j]] = (-sz)+maxDep[i[j]];
            //cout<<sz<<"\n";
            sz+=mp[{i[j],i[(j+1)%len]}];
        }
        auto it = s.end();it--;
        ans = max(ans,maxDep[i[0]]+sz+(*it).first);
        //cout<<(*it).second<<"\n";
        //cout<<ans<<endl;
        //cout<<i[0]<<" "<<i[1]<<"\n";
        long long global = 0;
        for(int j = i.size()-1;j>=1;j--){
            s.insert({(-global),i[(j+1)%len]});
            cost[i[(j+1)%len]] = (-global);
            global-=mp[{i[j],i[(j+1)%len]}];
            pair<long long,int> p = {cost[i[j]],i[j]};
            auto it=s.lower_bound(p);
            s.erase(it);
            it = s.end();it--;
            ans = max(ans,(maxDep[i[j]]+sz+(*it).first)+global);
            //cout<<(maxDep[i[j]]+sz+(*it).first)+global<<endl;
        }
        sum+=ans;
        //cout<<sum<<"\n";
    }
    cout<<sum<<endl;
}

Compilation message

islands.cpp: In function 'int main()':
islands.cpp:91:24: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   91 |         for(int j = 0;j<i.size();j++){
      |                       ~^~~~~~~~~
islands.cpp:122:24: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  122 |         for(int j = 1;j<i.size();j++){
      |                       ~^~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Correct 14 ms 31572 KB Output is correct
2 Incorrect 15 ms 31652 KB Output isn't correct
3 Correct 17 ms 31700 KB Output is correct
4 Correct 15 ms 31572 KB Output is correct
5 Correct 14 ms 31628 KB Output is correct
6 Correct 14 ms 31572 KB Output is correct
7 Incorrect 15 ms 31544 KB Output isn't correct
8 Incorrect 14 ms 31652 KB Output isn't correct
9 Correct 15 ms 31616 KB Output is correct
10 Correct 16 ms 31700 KB Output is correct
11 Correct 19 ms 31572 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 20 ms 31904 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 17 ms 32084 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 42 ms 35420 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 116 ms 46480 KB Output is correct
2 Incorrect 167 ms 58092 KB Output isn't correct
3 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 332 ms 80812 KB Output is correct
2 Correct 458 ms 95176 KB Output is correct
3 Incorrect 521 ms 119576 KB Output isn't correct
4 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 611 ms 130056 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Runtime error 676 ms 131072 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Runtime error 662 ms 131072 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -