AtCoder Grand Contest 010

Submission #1575393

Source codeソースコード

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int TAM = 100004;

vector <ll> adj [TAM];
ll val [TAM], pt [TAM];
int n;

void form( int x ){
    int nxt;
    for( int i = 0 ; i < adj[x].size() ; i ++ ){
        nxt = adj[x][i];
        if( nxt == pt[x] ) continue;
        pt[nxt] = x;
        //cout << " parent " << nxt << " = " << pt[nxt] << endl ;
        form(nxt);
    }
}

ll dn, up;
void solve ( ll cap , ll sum ){
    dn = sum - cap;
    up = cap - dn;
}

bool pos = 1;

ll find_up( int x ){
    //cout << " IN " << x << endl ;
    if( adj[x].size() == 1 ) return val[x];
    ll sum = 0, cap = val[x];
    int nxt;
    ll v;
    for( int i = 0 ; i < adj[x].size() ; i ++ ){
        nxt = adj[x][i];
        if( nxt == pt[x] ) continue;
        v = find_up(nxt);
        sum += v;
        if( v > cap ) pos = 0;
        //cout << x << " -> " << nxt << " = " << v << endl ;
    }
    solve( cap , sum );
    if( dn + up != cap or dn < 0 or up < 0 ) pos = 0;
    return up;
}

int main(){

    int a, b;
    scanf("%d",&n);
    for( int i = 1 ; i <= n ; i ++ )
        scanf("%lld",&val[i]);
    for( int i = 0 ; i < n-1 ; i ++ ){
        scanf("%d%d",&a,&b);
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    if( n == 2 ){
        if( val[1] == val[2] ) printf("YES\n");
        else printf("NO\n");
        return 0;
    }

    int rt;
    for( int i = 1 ; i <= n ; i ++ ){
        if( adj[i].size() > 1 ){
            pt [i] = i;
            form(i);
            rt = i;
            break;
        }
    }
    if( !find_up(rt) and pos ) printf("YES\n");
    else printf("NO\n");

    return 0;
}

Submission

Task問題 C - Cleaning
User nameユーザ名 vjudge1
Created time投稿日時
Language言語 C++14 (GCC 5.4.1)
Status状態 AC
Score得点 700
Source lengthソースコード長 1651 Byte
File nameファイル名
Exec time実行時間 69 ms
Memory usageメモリ使用量 12672 KB

Compiler messageコンパイルメッセージ

./Main.cpp: In function ‘int main()’:
./Main.cpp:51:19: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
scanf("%d",&n);
^
./Main.cpp:53:30: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
scanf("%lld",&val[i]);
^
./Main.cpp:55:28: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
scanf("%d%d",&a,&b);
^

Test case

Set

Set name Score得点 / Max score Cases
Sample - sample1.txt,sample2.txt,sample3.txt
All 700 / 700 sample1.txt,sample2.txt,sample3.txt,in1.txt,in10.txt,in11.txt,in12.txt,in13.txt,in14.txt,in15.txt,in16.txt,in17.txt,in18.txt,in19.txt,in2.txt,in20.txt,in21.txt,in22.txt,in23.txt,in24.txt,in25.txt,in26.txt,in27.txt,in28.txt,in29.txt,in3.txt,in30.txt,in31.txt,in32.txt,in33.txt,in34.txt,in35.txt,in36.txt,in37.txt,in38.txt,in39.txt,in4.txt,in40.txt,in41.txt,in42.txt,in43.txt,in44.txt,in45.txt,in5.txt,in6.txt,in7.txt,in8.txt,in9.txt,sample1.txt,sample2.txt,sample3.txt

Test case

Case name Status状態 Exec time実行時間 Memory usageメモリ使用量
in1.txt AC 52 ms 7936 KB
in10.txt AC 53 ms 7936 KB
in11.txt AC 54 ms 11648 KB
in12.txt AC 54 ms 11648 KB
in13.txt AC 51 ms 12672 KB
in14.txt AC 51 ms 11904 KB
in15.txt AC 2 ms 2560 KB
in16.txt AC 2 ms 2560 KB
in17.txt AC 2 ms 2560 KB
in18.txt AC 2 ms 2560 KB
in19.txt AC 52 ms 7936 KB
in2.txt AC 52 ms 7936 KB
in20.txt AC 53 ms 7936 KB
in21.txt AC 53 ms 7936 KB
in22.txt AC 52 ms 7936 KB
in23.txt AC 52 ms 7936 KB
in24.txt AC 69 ms 7936 KB
in25.txt AC 2 ms 2560 KB
in26.txt AC 56 ms 7936 KB
in27.txt AC 54 ms 7936 KB
in28.txt AC 2 ms 2560 KB
in29.txt AC 51 ms 7936 KB
in3.txt AC 52 ms 7936 KB
in30.txt AC 2 ms 2560 KB
in31.txt AC 56 ms 7936 KB
in32.txt AC 56 ms 7936 KB
in33.txt AC 52 ms 7936 KB
in34.txt AC 52 ms 7936 KB
in35.txt AC 52 ms 7936 KB
in36.txt AC 52 ms 7936 KB
in37.txt AC 3 ms 2688 KB
in38.txt AC 52 ms 7936 KB
in39.txt AC 53 ms 7936 KB
in4.txt AC 52 ms 7936 KB
in40.txt AC 52 ms 7936 KB
in41.txt AC 53 ms 7936 KB
in42.txt AC 52 ms 7936 KB
in43.txt AC 52 ms 7936 KB
in44.txt AC 53 ms 7936 KB
in45.txt AC 52 ms 7936 KB
in5.txt AC 54 ms 7936 KB
in6.txt AC 52 ms 7936 KB
in7.txt AC 49 ms 7296 KB
in8.txt AC 13 ms 3840 KB
in9.txt AC 53 ms 7936 KB
sample1.txt AC 2 ms 2560 KB
sample2.txt AC 2 ms 2560 KB
sample3.txt AC 2 ms 2560 KB