2019

2019

https://ac.nowcoder.com/acm/contest/1099/I

一个计数问题,下次看到数据就直接dp吧.有点分治思想在里面.
就是一个点一个点的计数,一条链一条链的来,f[i][j]表示到了i这个点值为j的方案数...如此转移下就好了.

#include <bits/stdc++.h>
using namespace std;
const int N=2e4+5,M=2e3+2e2;
int f[N][M];
struct vv{
    int to,val;
};
vector<vv>v[N];
int ans=0;

void dfs(int u,int fa)
{
    for(int i=0;i<2019;i++) f[u][i]=0;
    f[u][0]=1;
    for(int i=0;i<v[u].size();i++)
    {
        int x=v[u][i].to;int val=v[u][i].val;
        if(x==fa)   continue;
        dfs(x,u);
        for(int j=0;j<2019;j++)
        {
            ans+=f[u][j]*f[x][(4038-val-j)%2019];
        }
        for(int j=0;j<2019;j++)
        {
            f[u][(j+val)%2019]+=f[x][j];
        }
    }
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        ans=0;
        for(int i=1;i<=n;i++)   v[i].clear();
        for(int i=1;i<n;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            v[x].push_back({y,z});
            v[y].push_back({x,z});
        }
        dfs(1,-1);
        printf("%d\n",ans);
    }
    return 0;
}
lpt的小屋 文章被收录于专栏

我想要一份甜甜的爱情

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-24 13:32
点赞 评论 收藏
分享
认真搞学习:28小登的建议,投算法岗不要写什么物理竞赛,互联网+,多写点项目,用什么算法做了什么。还有本科算法是不可能的开发你这个也没有项目啊
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
6
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务