[HNOI2011]勾股定理

https://www.luogu.org/problemnew/show/P3213

首先要预处理出10^6以内的勾股数对。

我们有:(m^2-n^2)^2+(2mn)^2=(m^2+n^2)^2

枚举m,n就行了。

然后发现连出的图很稀疏,是一个森林上加几条非树边。

那么就对于每个连通块2^num枚举非树边较深的端点是否选,树形DP f[i][0/1]表示i号点是否选,子树的方案数。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1000005,md=1000000007;
vector<int> vct[N];
int n,i,j,k,l,a[N],head[N],adj[N],nxt[N],g[N],h[N],num,f[N][2],ans=1,w,v[N],cnt,dzx[N];
bool b[N];
inline void Read(int &x)
{
    char c;
    while((c=getchar())<'0'||c>'9');
    x=c-'0';
    while((c=getchar())>='0'&&c<='9')
        x=x*10+c-'0';
}
int gcd(int a,int b)
{
    if(!b)
        return a;
    return gcd(b,a%b);
}
inline void addedge(int u,int v)
{
    if(!a[u]||!a[v]||gcd(u,v)!=1)
        return;
    adj[++l]=v;
    nxt[l]=head[u];
    head[u]=l;
    adj[++l]=u;
    nxt[l]=head[v];
    head[v]=l;
    //printf("%d %d\n",u,v);
}
void dfs(int x,int dad)
{
    v[x]=++cnt;
    for(int y=head[x];y;y=nxt[y])
        if(adj[y]!=dad)
        {
            if(!v[adj[y]])
            {
                b[y]=true;
                dfs(adj[y],x);
            }
            else if(v[adj[y]]>v[x])//注意一条边不能被统计两次
            {
                ++num;
                g[num]=x,h[num]=adj[y];
                vct[x].push_back(adj[y]);
            }
        }
}
int pw(int a,int b)
{
    int rtn=1;
    while(b)
    {
        if(b&1)
            rtn=1ll*rtn*a%md;
        a=1ll*a*a%md;
        b>>=1;
    }
    return rtn;
}
void work(int x)
{
    int y;
    f[x][0]=1,f[x][1]=(pw(2,a[x])-1+md)%md;
    for(y=head[x];y;y=nxt[y])
        if(b[y])
        {
            work(adj[y]);
            f[x][0]=1ll*f[x][0]*(f[adj[y]][0]+f[adj[y]][1])%md;
            f[x][1]=1ll*f[x][1]*f[adj[y]][0]%md;
        }
    if(dzx[x]==1)
        f[x][0]=0;
    else if(dzx[x]==-1)
        f[x][1]=0;
    for(y=vct[x].size()-1;y>=0;--y)
        if(dzx[vct[x][y]]==1)
            break;
    if(y>=0)
        f[x][1]=0;
}
int main()
{
    Read(n);
    for(i=1;i<=n;++i)
    {
        Read(j);
        a[j]++;
    }
    for(i=1;i<=1000000;++i)
        for(j=i+1;2ll*i*j<=1000000;++j)
            if(1ll*j*j-1ll*i*i<=1000000)
                addedge(1ll*j*j-1ll*i*i,2ll*i*j);
    for(i=1;i<=1000000;++i)
        if(a[i]&&!v[i])
        {
            num=0;
            dfs(i,-1);
            w=0;
            for(j=0;j<(1<<num);++j)
            {
                for(k=1;k<=num;++k)
                    if((1<<(k-1))&j)
                        dzx[h[k]]=1;
                    else
                        dzx[h[k]]=-1;
                work(i);
                w=(1ll*w+f[i][0]+f[i][1])%md;
            }
            ans=1ll*ans*w%md;
        }
    printf("%d",(ans-1+md)%md);
    return 0;
}

 

转载于:https://www.cnblogs.com/pthws/p/11217920.html

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
根据引用\[1\]和引用\[2\]的描述,题目中的影魔拥有n个灵魂,每个灵魂有一个战斗力ki。对于任意一对灵魂对i,j (i<j),如果不存在ks (i<s<j)大于ki或者kj,则会为影魔提供p1的攻击力。另一种情况是,如果存在一个位置k,满足ki<c<kj或者kj<c<ki,则会为影魔提供p2的攻击力。其他情况下的灵魂对不会为影魔提供攻击力。 根据引用\[3\]的描述,我们可以从左到右进行枚举。对于情况1,当扫到r\[i\]时,更新l\[i\]的贡献。对于情况2.1,当扫到l\[i\]时,更新区间\[i+1,r\[i\]-1\]的贡献。对于情况2.2,当扫到r\[i\]时,更新区间\[l\[i\]+1,i-1\]的贡献。 因此,对于给定的区间\[l,r\],我们可以根据上述方法计算出区间内所有下标二元组i,j (l<=i<j<=r)的贡献之和。 #### 引用[.reference_title] - *1* *3* [P3722 [AH2017/HNOI2017]影魔(树状数组)](https://blog.csdn.net/li_wen_zhuo/article/details/115446022)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [洛谷3722 AH2017/HNOI2017 影魔 线段树 单调栈](https://blog.csdn.net/forever_shi/article/details/119649910)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值