LeetCode 1339.분열 두 갈래 나무의 최대 승적

6182 단어 LeetCode

제목


제목 링크

생각


먼저 전체 나무의 모든 노드 값을 합쳐서 계산하기;이어서 각 노드를 뿌리로 할 때 그 하위 나무의 노드 값sum의 합을 계산한다.이렇게 하면 하나의 노드와 하위 트리 노드의 값과res를 계산한 후에 이 노드로 트리를 두 개의 트리로 나누면 곱셈은res*(sum-res)이다.답을 얻을 때마다 전체 국면의 최우선치와 비교한다.나머지 문제에 대해res를long형으로 설정하고 마지막으로 출력할 때 모드 조작을 할 수 있다.
[귀속 흔히 볼 수 있는 유형] 1. 여기서 귀속은 반환 값을 가진 것이다. 반환 값을 가진 귀속은 만약에 당신이 큰 문제를 풀라고 한다면 먼저 작은 문제를 풀고 작은 문제의 답안으로 큰 문제를 해결해야 한다는 것을 이해할 수 있다.2. 또 하나는 전참귀속이다. 이런 유형의 귀속은 작은 문제가 큰 문제에 의존해야 하는 상황에 사용된다.

코드

class Solution {
public:
    int mod = 1e9 + 7;
    int sum = 0;
    long ans = 0;
    // 
    void DFS(TreeNode* root){
        if(root == NULL) return;
        sum += root -> val;
        DFS(root -> left);
        DFS(root -> right);
        return;
    }
    // 
    int subDFS(TreeNode* root){
        if(root == NULL) return 0;
        int left = subDFS(root -> left);
        int right = subDFS(root -> right);
        int res = left + right + root -> val;
        if( (long) res * (sum - res) > ans) ans = (long) res * (sum - res);
        return res;
    }
    int maxProduct(TreeNode* root) {
        DFS(root);
        subDFS(root);
        return ans % mod;
    }
};

좋은 웹페이지 즐겨찾기