Submission #779754

#	Time	Username	Problem	Language	Result	Execution time	Memory
779754		FEDIKUS	Graph (BOI20_graph)	C++17	0 / 100	2 ms	2652 KiB

Graph

#include<bits/stdc++.h>

using namespace std;

using ll = long long;

const ll maxn=1e5+10;

vector<pair<ll,ll> > g[maxn];
bool moze=true;
bool vis[maxn];
ll val[maxn];
ll sgn[maxn];
set<ll> a;

void dfs(ll node){
    vis[node]=true;
    for(auto i:g[node]){
        if(vis[i.first]){
            if(sgn[i.first]!=sgn[node] && val[i.first]+val[node]!=i.second) moze=false;
            if(sgn[i.first]==sgn[node]){
                if(sgn[i.first]==1) a.emplace(i.second-val[node]-val[i.first]);
                else a.emplace(val[i.first]+val[node]-i.second);
            }
        }else{
            sgn[i.first]=-sgn[node];
            val[i.first]=i.second-val[node];
            dfs(i.first);
        }
    }
}

int main(){
    ll n,m;
    cin>>n>>m;
    for(ll i=0;i<m;i++){
        ll u,v,c;
        cin>>u>>v>>c;
        g[u].push_back({v,c});
        g[v].push_back({u,c});
    }
    sgn[1]=1;
    val[1]=0;
    dfs(1);
    if(!moze || a.size()>1){
        cout<<"NO\n";
        return 0;
    }else cout<<"YES\n";
    if(a.size()==1){
        cout<<fixed<<showpoint<<setprecision(10);
        double aa=double(*a.begin())/2;
        for(ll i=1;i<=n;i++){
            cout<<aa*sgn[i]+val[i]<<" ";
        }
        return 0;
    }
    ll best=0;
    ll curr=LLONG_MAX;
    for(ll i=1;i<=n;i++){
        ll sum=0;
        for(ll j=1;j<=n;j++){
            sum+=abs(val[j]+val[i]*sgn[j]);
        }
        if(sum<curr){
            curr=sum;
            best=val[i];
        }
    }
    for(ll i=1;i<=n;i++){
        ll sum=0;
        for(ll j=1;j<=n;j++){
            sum+=abs(val[j]-val[i]*sgn[j]);
        }
        if(sum<curr){
            curr=sum;
            best=-val[i];
        }
    }
    for(ll i=1;i<=n;i++){
        cout<<val[i]+best*sgn[i]<<" ";
    }
    return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 12
• Subtask 3: 0 / 17
• Subtask 4: 0 / 24
• Subtask 5: 0 / 42