Submission #78929

#	Time	Username	Problem	Language	Result	Execution time	Memory
78929		Charis02	Pipes (BOI13_pipes)	C++14	72.78 / 100	1088 ms	62372 KiB

pipes

/*
solution is the same as described here: https://boi2013.informatik-olympiade.de/wp-content/uploads/2013/05/pipes-spoiler.pdf
*/

#include<iostream>
#include<stdio.h>
#include<vector>
#include<cmath>
#include<queue>
#include<string.h>
#include<map>
#include<set>
#include<algorithm>
#include<stack>
#define ll long long
#define pi pair < ll,ll >
#define mp(a,b) make_pair(a,b)
#define rep(i,a,b) for(int i = a;i < b;i++)
#define N 100004
#define INF 1e9+7

using namespace std;

ll n,m,ar[N],val[N],deg[N];
bool vis[N],cycle;
vector < vector < pi > > graph(N);
vector < pi > neo;
vector < pi > edges;

void solvetree(ll cur,ll par,ll edge)
{
    rep(i,0,graph[cur].size())
    {
        ll v = graph[cur][i].first;

        if(v == par)
            continue;

        solvetree(v,cur,graph[cur][i].second);

        ar[cur] -= val[graph[cur][i].second];
    }

//    cout << cur << " " << par << " " << edge << " " << ar[cur] <<endl;

    val[edge] = ar[cur];

    return;
}

int main()
{
    ios_base::sync_with_stdio(false);

    cin >> n >> m;

    rep(i,1,n+1)
    {
        cin >> ar[i];
    }

    rep(i,0,m)
    {
        ll a,b;
        cin >> a >> b;
        graph[a].push_back(mp(b,i));
        graph[b].push_back(mp(a,i));
        edges.push_back(mp(a,b));
    }

    if(m > n)
    {
        cout << 0 << endl;
        return 0;
    }
    else if(m == n-1)
    {
        solvetree(1,1,m);

        rep(i,0,m)
        {
            cout << 2*val[i] << endl;
        }
    }
    else
    {
        stack < pi > s;

        rep(i,1,n+1)
        {
            deg[i] = graph[i].size();

            if(deg[i] == 1)
            {
                vis[i] = true;
                s.push(mp(i,graph[i][0].second));
            }
        }

        while(!s.empty())
        {
            ll cur = s.top().first;
            ll e = s.top().second;
            s.pop();

            val[e] = ar[cur];

            rep(i,0,graph[cur].size())
            {
                ll v = graph[cur][i].first;

                if(vis[v])
                    continue;

                ar[v] -= val[e];

                deg[v]--;
                graph[v].erase(find(graph[v].begin(),graph[v].end(),mp(v,graph[v][i].second)));

                if(deg[v] == 1)
                {
                    s.push(mp(v,graph[v][0].second));
                }
            }
        }

        ll st = 1;

        while(vis[st])
        {
            st++;
        }

        ll start = st;

        while(!vis[start])
        {
            rep(i,0,graph[st].size())
            {
                ll v=  graph[st][i].first;

                if(!vis[v])
                {
                    neo.push_back(mp(st,graph[st][i].second));
                    vis[v] = true;
                    st = v;
                    break;
                }
            }
        }

    /*    cout << neo.size() << endl;

        rep(i,0,neo.size())
        {
            cout << neo[i].first <<  " " << ar[neo[i].first] << endl;
        }
*/
        if((neo.size())%2==0)
        {
            cout << 0 << endl;
        }
        else
        {
            ll s = 0;

            rep(i,0,neo.size())
            {
                s += ar[neo[i].first]*pow(-1,i%2);
            }

            val[neo[neo.size()-1].second] = s/2;
            val[neo[0].second] = ar[neo[0].first] - s/2;

            rep(i,1,neo.size()-1)
            {
                val[neo[i].second] =ar[neo[i].first]  - val[neo[i-1].second];
            }

            rep(i,0,m)
            {
                cout << 2*val[i] << endl;
            }
        }
    }

    return 0;
}

/*
3 3
1 3 2
1 2
1 3
2 3

4 4
1 3 2 -4
1 2
4 3
2 3
2 4
*/

Compilation message (stderr)

pipes.cpp: In function 'void solvetree(long long int, long long int, long long int)':
pipes.cpp:18:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
pipes.cpp:32:9:
     rep(i,0,graph[cur].size())
         ~~~~~~~~~~~~~~~~~~~~~       
pipes.cpp:32:5: note: in expansion of macro 'rep'
     rep(i,0,graph[cur].size())
     ^~~
pipes.cpp: In function 'int main()':
pipes.cpp:18:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
pipes.cpp:108:17:
             rep(i,0,graph[cur].size())
                 ~~~~~~~~~~~~~~~~~~~~~
pipes.cpp:108:13: note: in expansion of macro 'rep'
             rep(i,0,graph[cur].size())
             ^~~
pipes.cpp:18:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
pipes.cpp:138:17:
             rep(i,0,graph[st].size())
                 ~~~~~~~~~~~~~~~~~~~~
pipes.cpp:138:13: note: in expansion of macro 'rep'
             rep(i,0,graph[st].size())
             ^~~
pipes.cpp:18:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
pipes.cpp:167:17:
             rep(i,0,neo.size())
                 ~~~~~~~~~~~~~~      
pipes.cpp:167:13: note: in expansion of macro 'rep'
             rep(i,0,neo.size())
             ^~~
pipes.cpp:18:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i,a,b) for(int i = a;i < b;i++)
pipes.cpp:175:17:
             rep(i,1,neo.size()-1)
                 ~~~~~~~~~~~~~~~~    
pipes.cpp:175:13: note: in expansion of macro 'rep'
             rep(i,1,neo.size()-1)
             ^~~

• Subtask 1: 30 / 30
• Subtask 2: 42.77777777777778 / 70