Ybtoj-排列计数【矩阵乘法,分块幂】
生活随笔
收集整理的這篇文章主要介紹了
Ybtoj-排列计数【矩阵乘法,分块幂】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
正題
題目鏈接:http://noip.ybtoj.com.cn/contest/596/problem/1
題目大意
TTT組詢問給出nnn求有多少個nnn的排列滿足第一個是111并且相鄰的差不超過222。
1≤T≤106,1≤n≤1091\leq T\leq 10^6,1\leq n\leq 10^91≤T≤106,1≤n≤109
解題思路
考慮一下如果我們要不斷向前填滿前面的一段的話,那么填的方案就只有兩種,(x,x+1)(x,x+1)(x,x+1)和(x,x+2,x+1,x+3)(x,x+2,x+1,x+3)(x,x+2,x+1,x+3),這樣一下會跳111或者333。
然后還有一種方法是一直往前跳兩格然后再跳回來,但是這樣就直接結束了。
設fif_ifi?表示按照最前面那兩種方法鋪iii個的方案,那么答案就是(∑i=1nfi)?fn?1(\sum_{i=1}^nf_{i})-f_{n-1}(∑i=1n?fi?)?fn?1?,減去n?1n-1n?1是因為會被fnf_nfn?算重。
然后矩陣乘法+分塊預處理光速冪做就好了。
時間復雜度O(43(109+T))O(4^3(\sqrt {10^9}+T))O(43(109?+T))
code
#include<cstdio> #include<cstring> #include<algorithm> #include<vector> #define ll long long #define file(x) freopen("data"#x".in","r",stdin);freopen("data"#x".out","w",stdout); using namespace std; const ll N=5e5+10; struct node{ll to,next; }a[N<<1]; ll n,m,tot=1,ls[N],fa[N],dep[N],s[N],f[N],ans,num; vector<ll> T[N]; void addl(ll x,ll y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;return; } void dfs(ll x,ll from){dep[x]=dep[fa[x]]+1;for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(i==from)continue;else if(dep[y]){if(dep[y]<=dep[x])s[x]++,s[fa[y]]--;continue;}T[x].push_back(y);fa[y]=x;dfs(y,i^1);s[x]+=s[y];}num+=(s[x]==0);return; } void calc(ll x){f[x]=(s[x]==0);dep[x]=1;for(ll i=0;i<T[x].size();i++){ll y=T[x][i];calc(y);ans+=f[x]*dep[y]+f[y]*dep[x];dep[x]+=dep[y];f[x]+=(s[x]==0)*dep[y]+f[y];}return; } signed main() {file(2);scanf("%lld%lld",&n,&m);for(ll i=1;i<=m;i++){ll x,y;scanf("%lld%lld",&x,&y);addl(x,y);addl(y,x);}dfs(1,0);calc(1);printf("%lld\n",n*(n-1)/2ll*num-ans);return 0; }總結
以上是生活随笔為你收集整理的Ybtoj-排列计数【矩阵乘法,分块幂】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 天玑 9300 搭载联发科第二代硬件光线
- 下一篇: 关于爱情的唯美句子