Cod sursa(job #1759785)

Utilizator mariakKapros Maria mariak Data 19 septembrie 2016 20:28:52
Problema Zvon Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.98 kb
#include <bits/stdc++.h>
#define Nmax 100002
#define pb(x) push_back(x)

FILE *fout = freopen("zvon.out", "w", stdout);

using namespace std;
int T, n, dpth[Nmax], k[Nmax];
bool ok;
vector <int> G[Nmax];
class InputReader
{
public:
    InputReader() {}
    InputReader(const char *file_name)
    {
        input_file = fopen(file_name, "r");
        cursor = 0;
        fread(buffer, SIZE, 1, input_file);
    }
    inline InputReader &operator >>(int &n)
    {
        while(buffer[cursor] < '0' || buffer[cursor] > '9')
        {
            advance();
        }
        n = 0;
        while('0' <= buffer[cursor] && buffer[cursor] <= '9')
        {
            n = n * 10 + buffer[cursor] - '0';
            advance();
        }
        return *this;
    }
private:
    FILE *input_file;
    static const int SIZE = 1 << 17;
    int cursor;
    char buffer[SIZE];
    inline void advance()
    {
        ++ cursor;
        if(cursor == SIZE)
        {
            cursor = 0;
            fread(buffer, SIZE, 1, input_file);
        }
    }
};
void initialize()
{
    for(int i = 1; i < n; ++ i)
        G[i].clear();
    ok = 0;
    memset(dpth, 0, sizeof(dpth));
    memset(k, 0, sizeof(k));
}
int DFS(int x)
{
    int maxx = 0;
    if(!G[x].size()) k[x] = 1;
    for(int i = 0; i < G[x].size(); ++ i)
    {
        int y = G[x][i];
        DFS(y);
        if(dpth[x] == dpth[y] + 1) k[x] += k[y];
        if(dpth[x] < dpth[y] + 1)
            dpth[x] = dpth[y] + 1, k[x] = k[y];
    }
}
int main()
{
    int x, y;
    InputReader cin("zvon.in");
    cin >> T;
    while(T --)
    {
        initialize();
        cin >> n;
        if(n == 1)
            ok = 1;

        for(int i = 1; i < n; ++ i)
        {
            cin >> x >> y;
            G[x].pb(y);
        }
        if(!ok)
        {
            DFS(1);
            printf("%d\n", dpth[1] + k[1] - 1);
        }
        else printf("0\n");
    }
    return 0;
}