트리 dp

1482 단어
//hdu 5593   dp
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <vector>
#define LL long long
#define maxn 500005
using namespace std;
int dp[maxn][15];
int k,n;
vector<int> g[maxn];

void dfs(int a)
{
    dp[a][0]=1;
    for(int i=0;i<g[a].size();i++)
    {
        int j=g[a][i];
        dfs(j);
        for(int ii=1;ii<=k;ii++)
           dp[a][ii]+=dp[j][ii-1];
    }
}

void dfs_(int a)
{
    for(int i=0;i<g[a].size();i++)
    {
        int j=g[a][i];
        for(int ii=k;ii>=2;ii--)
            dp[j][ii]+=dp[a][ii-1]-dp[j][ii-2];
        dp[j][1]+=dp[a][0];
        dfs_(j);
    }
}

int main()
{
    //freopen("d:\\in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T)
    {
        int a,b;
        for(int i=1;i<=maxn;i++)
            g[i].clear();
        memset(dp,0,sizeof(dp));
        scanf("%d%d%d%d",&n,&k,&a,&b);
        for(int i=2;i<=n;i++)
        {
            int cnt;
            cnt=((LL)a*i+b)%(i-1)+1;
            g[cnt].push_back(i);
        }
        dfs(1);
        dfs_(1);
        long long int ans=0;
        for(int i=1;i<=n;i++)
        {
            long long int temp=0;
            for(int ii=0;ii<=k;ii++)
            temp+=dp[i][ii];
            ans^=temp;
        }
        printf("%lld
",ans); T--; } return 0; }

좋은 웹페이지 즐겨찾기