회원가입
로그인
Toggle navigation
문제
문제
전체 문제
문제 출처
단계별로 풀어보기
알고리즘 분류
추가된 문제
문제 순위
문제
푼 사람이 한 명인 문제
아무도 못 푼 문제
최근 제출된 문제
최근 풀린 문제
랜덤
출처
ICPC
Olympiad
한국정보올림피아드
한국정보올림피아드시․도지역본선
전국 대학생 프로그래밍 대회 동아리 연합
대학교 대회
카카오 코드 페스티벌
Coder's High
ICPC
Regionals
World Finals
Korea Regional
Africa and the Middle East Regionals
Europe Regionals
Latin America Regionals
North America Regionals
South Pacific Regionals
문제집
대회
2
채점 현황
랭킹
게시판
그룹
더 보기
재채점 기록
블로그
강의
실험실
도움말
BOJ Stack
BOJ Book
전체
공지
자유
질문
오타/오역/요청
게시판 공지
홍보
업데이트
solved.ac
글쓰기
질문 도움말
자주묻는 질문
재귀로 풀어서 틀렸습니다.
1289번 - 트리의 가중치
again7536
4년 전
0
제가 케이스를 만들어서 몇개 넣어봤는데 그건 잘 나오네요.
(완전 이진 트리, child가 많은 트리 등)
예외 케이스를 아무리 생각해도 안나오는데 어떤게 있을까요?
#include <iostream> #include <vector> #include <cstring> #define MAX 1000000007 using namespace std; long long int calweight(int v, int root, int parent, long long int& net); typedef struct { int vertex; long long int weight; }node; vector<node> tree[100000]; int main(void) { int v; cin >> v; for (int i = 0; i < v - 1; i++) { int a; cin >> a; node n, m; cin >> n.vertex >> n.weight; m.vertex = a; m.weight = n.weight; tree[a].push_back(n); tree[n.vertex].push_back(m); } long long int net = 0; calweight(v, 1, -1, net); cout << net; return 0; } long long int calweight(int v, int root, int parent, long long int& net) { long long int ret = 0; int size = tree[root].size(); long long int* subweight = new long long int[size]; memset(subweight, 0, size*sizeof(long long int)); for (int i = 0; i < size; i++) { if (tree[root][i].vertex == parent) continue; subweight[i] = calweight(v, tree[root][i].vertex, root, net) % MAX; } for (int i = 0; i < size; i++) { if (tree[root][i].vertex == parent) continue; //ret counts the path ends at root. ret += tree[root][i].weight % MAX; ret += tree[root][i].weight * subweight[i] % MAX; for (int j = i + 1; j < size; j++) { //net counts the path does not end at root. if (tree[root][j].vertex == parent) continue; net += subweight[i] * subweight[j] * tree[root][i].weight * tree[root][j].weight % MAX; net += tree[root][i].weight * tree[root][j].weight % MAX; net += subweight[i] * tree[root][j].weight * tree[root][i].weight % MAX; net += tree[root][i].weight * tree[root][j].weight * subweight[j] % MAX; } } net += ret % MAX; net = net % MAX; delete [] subweight; return ret; }
seungchan0325
9달 전
0
옛날에 글을 남기신거라 보실진 모르겠지만, tree의 크기가 1 더 커야합니다.
댓글을 작성하려면
로그인
해야 합니다.
again7536 4년 전
제가 케이스를 만들어서 몇개 넣어봤는데 그건 잘 나오네요.
(완전 이진 트리, child가 많은 트리 등)
예외 케이스를 아무리 생각해도 안나오는데 어떤게 있을까요?