結果

問題 No.439 チワワのなる木
ユーザー ytftytft
提出日時 2021-03-26 07:04:39
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 121 ms / 5,000 ms
コード長 4,421 bytes
コンパイル時間 2,800 ms
コンパイル使用メモリ 192,392 KB
実行使用メモリ 32,800 KB
最終ジャッジ日時 2024-05-05 17:32:42
合計ジャッジ時間 4,329 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 1 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 3 ms
5,376 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 89 ms
23,872 KB
testcase_19 AC 82 ms
22,496 KB
testcase_20 AC 117 ms
29,976 KB
testcase_21 AC 34 ms
11,352 KB
testcase_22 AC 30 ms
10,440 KB
testcase_23 AC 121 ms
31,784 KB
testcase_24 AC 118 ms
32,420 KB
testcase_25 AC 111 ms
31,400 KB
testcase_26 AC 101 ms
31,140 KB
testcase_27 AC 109 ms
32,800 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function 'std::vector<std::vector<int> > Graph<verticesValueType, edgesValueType>::getConnected() [with verticesValueType = int; edgesValueType = int]':
main.cpp:30:5: warning: control reaches end of non-void function [-Wreturn-type]
   30 |     }
      |     ^

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

template<typename verticesValueType,typename edgesValueType>
class Graph{
public:
    vector<vector<int>> edges;
    vector<int> parent;
    int numberOfVertices;
    vector<vector<int>> adjacencyMatrix;
    vector<vector<int>> connected;
    vector<vector<int>> children;
    vector<verticesValueType> verticesValue;
    vector<edgesValueType> edgesValue;
    Graph(vector<vector<int>> e,int N){
        edges=e;
        numberOfVertices=N;
    }
    vector<vector<int>> getConnected(){
        if(connected.size()>0)return connected;
        if(edges.size()>0){
            connected.resize(numberOfVertices);
            connected.clear();
            for(int i=edges.size()-1;i>=0;i--){
                connected[edges[i][0]].push_back(edges[i][1]);
                connected[edges[i][1]].push_back(edges[i][0]);
            }
            return connected;
        }
    }
    vector<int> getParent(int root=0){
        if(parent.size()>0)return parent;
        parent.resize(numberOfVertices);
        vector<bool> isWritten(numberOfVertices);
        vector<vector<int>> c=getConnected();
        queue<vector<int>> q;
        q.push({-1,root});
        while(!q.empty()){
            if(isWritten[q.front()[1]]){
                goto end;
            }
            isWritten[q.front()[1]]=true;
            parent[q.front()[1]]=q.front()[0];
            for(int i:connected[q.front()[1]]){
                q.push({q.front()[1],i});
            }
            end:{}
            q.pop();
        }
        return parent;
    }
    vector<vector<int>> getChildren(int root=0){
        getParent(root);
        if(children.size()>0)return children;
        children.resize(numberOfVertices);
        children.clear();
        for(int i=0;i<numberOfVertices;i++){
            if(parent[i]>-1){
                children[parent[i]].push_back(i);
            }
        }
        return children;
    }
    vector<int> orderFromChildren(){
        getParent();
        getChildren();
        vector<int> ret(0);
        vector<int> count(numberOfVertices);
        queue<int> q;
        for(int i=0;i<numberOfVertices;i++){
            if(children[i].size()==0){
                q.push(i);
            }
        }
        while(true){
            ret.push_back(q.front());
            if(parent[q.front()]==-1){
                return ret;
            }
            count[parent[q.front()]]++;
            if(count[parent[q.front()]]==children[parent[q.front()]].size()){
                q.push(parent[q.front()]);
            }
            q.pop();
        }
        return ret;
    }
};
int main(){
    int N;
    string S;
    cin>>N>>S;
    vector<vector<int>> edges(0,vector<int>(2));
    int a,b;
    for(int i=0;i<N;i++){
        cin>>a>>b;
        edges.push_back({a-1,b-1});
    }
    Graph<int,int> g(edges,N);
    vector<int> isC(N);
    for(int i=0;i<N;i++){
        isC[i]=(S[i]=='c');
    }
    vector<int> ord=g.orderFromChildren();
    vector<long long> c(N),w(N);
    for(int i:ord){
        c[i]+=isC[i];
        w[i]+=(1-isC[i]);
        if(g.parent[i]!=-1){
            c[g.parent[i]]+=c[i];
            w[g.parent[i]]+=w[i];
        }
    }
    vector<long long> cw(N),ww(N),pcw(N),pww(N);
    for(int i:ord){
        if(!isC[i]){
            cw[i]+=c[i];
            ww[i]+=w[i]-1;
            pcw[i]=c[i];
            pww[i]=w[i]-1;
        }
        if(g.parent[i]!=-1){
            cw[g.parent[i]]+=cw[i];
            ww[g.parent[i]]+=ww[i];
        }
    }
    vector<long long> cww(N);
    for(int i:ord){
        if(isC[i]){
            cww[i]+=ww[i];
            cww[i]+=w[i]*cw[i];
            for(int j:g.children[i]){
                cww[i]-=w[j]*cw[j];
            }
            cww[i]+=(c[i]-1)*ww[i];
            for(int j:g.children[i]){
                cww[i]-=c[j]*ww[j];
            }
        }else{
            cww[i]+=cw[i]-pcw[i];
            cww[i]+=c[i]*(w[i]-1);
            for(int j:g.children[i]){
                cww[i]-=c[j]*w[j];
            }
            cww[i]+=(w[i]-1)*(cw[i]-pcw[i]);
            for(int j:g.children[i]){
                cww[i]-=w[j]*cw[j];
            }
            cww[i]+=c[i]*(ww[i]-pww[i]);
            for(int j:g.children[i]){
                cww[i]-=c[j]*ww[j];
            }
        }
        if(g.parent[i]!=-1){
            cww[g.parent[i]]+=cww[i];
        }
    }
    cout<<cww[0]<<endl;
}
0