关闭

[二维树状数组]BZOJ 1452—— [JSOI2009]Count

标签: BZOJ树状数组
322人阅读 评论(0) 收藏 举报
分类:

题目梗概

给出一个N*M的方格,每个方格有一个数字。

每次更改某个格子上的数字或询问一个子矩阵的某值的个数。

解题思路

裸的二维树状数组,对每个值分别开数组记录就可以了。

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=25,maxm=125,tt=45989;
struct jz{
    int x[maxm][maxm],n,m;
    jz operator*(const jz &b)const{
        jz c;memset(c.x,0,sizeof(c.x));
        c.n=n,c.m=b.m;
        for (int i=1;i<=c.n;i++)
        for (int j=1;j<=c.m;j++)
        for (int k=1;k<=b.n;k++)
        c.x[i][j]=(c.x[i][j]+x[i][k]*b.x[k][j])%tt;
        return c;
    }
}a,b;
int n,m,t,A,B,tot=1,nxt[maxm],son[maxm],lnk[maxn],ans;
void add(int x,int y){
    nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;
}
void work(){
    for (int i=0;i<n;i++)
    for (int j=lnk[i];j;j=nxt[j])
    for (int k=lnk[son[j]];k;k=nxt[k])
    if (k!=(j^1)) b.x[j][k]++;
}
jz qsm(jz w,int b){
    jz c;c.n=w.n;c.m=w.m;
    for (int i=1;i<=c.n;i++)
    for (int j=1;j<=c.m;j++) c.x[i][j]=(i==j);
    while (b>0){
        if (b%2==1) c=c*w;
        w=w*w;
        b=b>>1;
    }
    return c;
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    scanf("%d%d%d%d%d",&n,&m,&t,&A,&B);
    for (int i=1;i<=m;i++){
        int x,y;scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    work();
    for (int j=lnk[A];j;j=nxt[j]) a.x[1][j]++;
    a.n=1;b.n=a.m=b.m=tot;
    a=a*qsm(b,t-1);
    for (int j=lnk[B];j;j=nxt[j]) ans=(ans+a.x[1][j^1])%tt;
    printf("%d\n",ans);
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:39610次
    • 积分:1709
    • 等级:
    • 排名:千里之外
    • 原创:123篇
    • 转载:1篇
    • 译文:0篇
    • 评论:46条
    我的日记
    博客专栏