【HDU】5279 YJC plays Minecraft【cdq分治+NTT】

原创 2015年07月09日 11:48:18

传送门:【HDU】5279 YJC plays Minecraft

忘记修改测试时的分治大小交了一发,不然就1A了……

整个题目的关键在对N个点的森林个数计数,这个我是昨晚躺床上突然想到的www,不过队友早就过了……膜拜之。dp一直是我的弱点啊= =

fn为n个点的生成树个数,幸运的是fn=nn2

则对于n个点的森林个数,有公式:

dpn=i=1ndpniCi1n1fi

对于n个点的1和n在同一棵生成树的森林个数,有公式:
dp2n=i=2ndpniCi2n2fi

上面的式子转换一下就可以变成卷积形式,然后我们用cdq分治来加速NTT过程。预处理复杂度O(Nlog2N)

则对于给定的数据,答案为

2ni=1ndp[ai]i=ndp2[ai]

my  code:

/*************************************************************************
    > File Name: 5279.cpp
    > Author: poursoul
    > Created Time: 2015年07月09日 星期四 08时50分42秒
 ************************************************************************/

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std ;

typedef long long LL ;

#define clr(a,x); memset ( a , x , sizeof a ) ;

const int MAXN = 1 << 17 ;
const int mod = 998244353 ;
const int g = 3 ;

LL f[MAXN] , invf[MAXN] ;
LL tcnt[MAXN] ;
LL dp[MAXN] , dp2[MAXN] ;
LL x1[MAXN] , x2[MAXN] , x3[MAXN] ;
int n ;

LL power ( LL a , int b ) {
    LL res = 1 , tmp = a ;
    while ( b ) {
        if ( b & 1 ) res = res * tmp % mod ;
        tmp = tmp * tmp % mod ;
        b >>= 1 ;
    }
    return res ;
}

void NTT ( LL y[] , int n , int rev ) {
    for ( int i = 1 , j , k , t ; i < n ; ++ i ) {
        for ( j = 0 , k = n >> 1 , t = i ; k ; k >>= 1 , t >>= 1 ) {
            j = j << 1 | ( t & 1 ) ;
        }
        if ( i < j ) swap ( y[i] , y[j] ) ;
    }
    for ( int s = 2 , ds = 1 ; s <= n ; ds = s , s <<= 1 ) {
        LL wn = power ( g , ( mod - 1 ) / s ) ;
        if ( !rev ) wn = power ( wn , mod - 2 ) ;
        for ( int k = 0 ; k < n ; k += s ) {
            LL w = 1 , t ;
            for ( int i = k ; i < k + ds ; ++ i , w = w * wn % mod ) {
                y[i + ds] = ( y[i] - ( t = y[i + ds] * w % mod ) + mod ) % mod ;
                y[i] = ( y[i] + t ) % mod ;
            }
        }
    }
    if ( !rev ) {
        LL invn = power ( n , mod - 2 ) ;
        for ( int i = 0 ; i < n ; ++ i ) {
            y[i] = y[i] * invn % mod ;
        }
    }
}

void cdq_fz ( int l , int r ) {
    if ( l == r ) return ;
    if ( l + 1 == r ) {
        dp[r] = ( dp[r] + dp[l] ) % mod ;
        return ;
    }
    int m = ( l + r ) >> 1 , n1 = 1 ;
    cdq_fz ( l , m ) ;
    while ( n1 <= r - l + 1 ) n1 <<= 1 ;
    for ( int i = 0 ; i < n1 ; ++ i ) {
        x1[i] = l + i <= m ? dp[l + i] * invf[l + i] % mod : 0 ;
        x2[i] = l + i <= r ? tcnt[i + 1] * invf[i] % mod : 0 ;
        x3[i] = l + i <= r ? tcnt[i + 2] * invf[i] % mod : 0 ;
    }
//  if ( m + 1 <= 2 && 2 <= r ) printf ( "%lld %d\n" , x3[0] * x1[0] , n1 ) ;
    NTT ( x1 , n1 , 1 ) ;
    NTT ( x2 , n1 , 1 ) ;
    NTT ( x3 , n1 , 1 ) ;
    for ( int i = 0 ; i < n1 ; ++ i ) {
        x3[i] = x1[i] * x3[i] % mod ;
        x2[i] = x1[i] * x2[i] % mod ;
    }
    NTT ( x2 , n1 , 0 ) ;
    NTT ( x3 , n1 , 0 ) ;
//  if ( m + 1 <= 2 && 2 <= r ) printf ( "%d %d %lld %lld\n" , l , r , x3[0], dp[0] * invf[0] *  tcnt[2] ) ;
    for ( int i = m + 1 ; i <= r ; ++ i ) {
        dp[i] = ( dp[i] + f[i - 1] * x2[i - l - 1] % mod ) % mod ;
        dp2[i] = ( dp2[i] + f[i - 2] * x3[i - l - 2] % mod ) % mod ;
    }
    cdq_fz ( m + 1 , r ) ;
}

void preprocess () {
    f[0] = invf[0] = 1 ;
    dp[0] = dp2[0] = dp2[1] = 1 ;
    for ( int i = 1 ; i <= 100000 ; ++ i ) {
        f[i] = f[i - 1] * i % mod ;
        invf[i] = power ( f[i] , mod - 2 ) ;
        if ( i >= 2 ) tcnt[i] = power ( i , i - 2 ) ;
    }
    tcnt[1] = 1 ;
    cdq_fz ( 0 , 100000 ) ;
//  for ( int i = 0 ; i <= 10 ; ++ i ) {
//      printf ( "%lld %lld\n" , dp[i] , dp2[i] ) ;
//  }
}

void solve () {
    int x ;
    scanf ( "%d" , &n ) ;
    int ans1 = power ( 2 , n ) , ans2 = 1 ;
    for ( int i = 1 ; i <= n ; ++ i ) {
        scanf ( "%d" , &x ) ;
        ans1 = ans1 * dp[x] % mod ;
        if ( n > 2 ) ans2 = ans2 * dp2[x] % mod ;
    }
//  printf ( "%d %d\n" , ans1 , ans2 ) ;
    printf ( "%d\n" , ( ans1 - ans2 + mod ) % mod ) ;
}

int main () {
    int T ;
    preprocess () ;
    scanf ( "%d" , &T ) ;
    for ( int i = 1 ; i <= T ; ++ i ) {
        solve () ;
    }
    return 0 ;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

hdu 3335 Divisibility(Dancing Links 重复覆盖)

Divisibility Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) T...
  • u011699990
  • u011699990
  • 2015年05月10日 00:30
  • 484

HDU 5909 Tree Cutting 树形DP+快速沃尔什变换

题目大意:给出一棵树,每个点有一个点权,求对于每个i∈[0,m)i\in[0,m)输出有多少个连通诱导子图的异或和为ii n≤1000n\leq1000,m...
  • PoPoQQQ
  • PoPoQQQ
  • 2016年10月14日 16:11
  • 1739

HDU_4069_SquigglySudoku(DancingLinksX精确覆盖+BFS)

Squiggly Sudoku Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)...
  • baidu_29410909
  • baidu_29410909
  • 2016年05月01日 10:52
  • 150

HDU 4069 Squiggly Sudoku Dancing-Links(DLX)+Floodfill

题目大意:。。还是数独,不同的是原先的九宫格约束条件变为了给定的任意形状。。。 我们跑一遍floodfill 得出每一个格子属于哪一个形状 然后就是裸的数独了 这题T 好不容易改完了 尼玛 交...
  • PoPoQQQ
  • PoPoQQQ
  • 2014年09月04日 17:38
  • 1428

[DLX精确覆盖] hdu 4210 Su-domino-ku

题意: 首先给你N块板(2*1或1*2)的,数字以及位置 再给你9个单个的板,1~9,分别的位置 然后问你如何放其他板(1*2或2*1)使得成为数独 注意的是对于板子来说,含有的数字是唯一的 ...
  • wdcjdtc
  • wdcjdtc
  • 2015年07月22日 15:54
  • 715

HDU_4979_ASimpleMathProblem.(DancingLinksX重复覆盖)

A simple math problem. Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/O...
  • baidu_29410909
  • baidu_29410909
  • 2016年05月02日 20:53
  • 300

ZOJ_3029_TreasureMap(DancingLinksX精确覆盖)

Treasure Map Time Limit: 2 Seconds      Memory Limit: 32768 KB Your boss once had got many copies...
  • baidu_29410909
  • baidu_29410909
  • 2016年04月30日 23:47
  • 210

HDU 4757 Tree (倍增算法求LCA + 可持久化Trie树)

题目大意: 就是现在给出一棵树, 结点个数不超过10W, 每个节点上有一个不超过2^16的非负整数, 然后10W次询问, 每次询问两个节点的路径上的所有数中异或上给出的数的最大值 大致思路...
  • u013738743
  • u013738743
  • 2015年10月05日 21:58
  • 784

[搜索] ICPC 2016 Qingdao J & HDU 5991 Cliques

给你一个图,你可以执行删边或者加边这样的操作,然后将这个图变成若干个团,问最小的步数。 如果答案超过10,输出-1。dls orzz 想偷懒用set 结果T了#include #include #...
  • u014609452
  • u014609452
  • 2017年03月29日 07:47
  • 495

【HDU】4848 Wow! Such Conquering! dfs爆搜+强力剪枝

Wow! Such Conquering! Time Limit: 15000/8000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/...
  • u013368721
  • u013368721
  • 2014年07月09日 11:29
  • 901
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:【HDU】5279 YJC plays Minecraft【cdq分治+NTT】
举报原因:
原因补充:

(最多只允许输入30个字)