85分蜜汁WA三个点求助

RT，无论是在 luogu 还是 loj 交了一遍又一遍，14,17,18 三个点总是 \rm{WA} 掉，不知道为什么，瞎改了很多地方这三个点还是不断的 \rm{WA} ，故求助大佬。

二分的地方自认为很迷，但是不知道应该怎么改才好......尝试过重构但未果。

Code:

#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;

const int N=6e5+2;
const ll inf=1e18+9;

int n,k,head[N],cnt;
struct date {
    ll x;int y;
    bool operator < (const date&var) const {return x==var.x?y>var.y:x<var.x;}
    date operator + (const date&var) {return (date){x+var.x,y+var.y};}
    date operator + (const ll&var) {return (date){x+var,y};}
}dp[3][N];
ll number(date var) {return var.x;}
struct Edge{int nxt,to,val;} G[N<<1];
void add(int u,int v,int w) {
    G[++cnt]=(Edge){head[u],v,w},head[u]=cnt;
    G[++cnt]=(Edge){head[v],u,w},head[v]=cnt;
}

template <typename _Tp> inline void IN(_Tp&x) {
    char ch;bool flag=0;x=0;
    while(ch=getchar(),!isdigit(ch)) if(ch=='-') flag=1;
    while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    if(flag) x=-x;
}

void check(ll add,int u,int fa) {
    dp[0][u]=(date){0,0},
    dp[1][u]=(date){-inf,0},
    dp[2][u]=(date){add,1};
    for(int e=head[u];e;e=G[e].nxt) {
        int v=G[e].to;
        ll w=G[e].val;
        if(v==fa) continue;
        check(add,v,u);
        date num=max(dp[0][v],max(dp[1][v],dp[2][v]));

        dp[2][u]=max(dp[2][u],dp[2][u]+num);
        dp[2][u]=max(dp[2][u],dp[1][u]+(date){w,0}+max(dp[0][v],dp[1][v]+(date){-add,-1}));

        dp[1][u]=max(dp[1][u],dp[1][u]+num);
        dp[1][u]=max(dp[1][u],dp[0][u]+(date){w,0}+max(dp[1][v],dp[0][v]+(date){add,1}));

        dp[0][u]=dp[0][u]+num;
    }return;
}

ll wqs(ll sum) {
    ll l=-sum,r=sum,mid;
    date now;
    while(l<r) {
        mid=(l+r+1)>>1,check(mid,1,0);
        now=max(dp[0][1],max(dp[1][1],dp[2][1]));
        if(!(now.y^k)) {l=r=mid;break;}
        now.y<k?l=mid:r=mid-1;
    }
    check(l,1,0);
    now=max(dp[0][1],max(dp[1][1],dp[2][1]));
    return now.x-1ll*mid*k;
}

int main() {
    // freopen("lct2.in","r",stdin);
    // freopen("P4383.out","w",stdout);
    IN(n),IN(k);++k;
    ll sum=0;
    for(int i=1;i<n;++i) {
        int x,y,v;IN(x),IN(y),IN(v);
        add(x,y,v),sum+=abs(v);
    }
    printf("%lld\n",wqs(sum));
    return 0;
}