学习笔记1:图论解决数学问题

E - Non-Decreasing Colorful Path (atcoder.jp)icon-default.png?t=N7T8https://atcoder.jp/contests/abc335/tasks/abc335_e

思路:

数值低的点向数值高的点dp

解法1:并查集判断连通性(如果在同一块中则表示数值相等),自定义排序(先低后高),dp处理(dp初始化时,1所在的块为1,其他为负无穷),答案是点n所在的块

#include <bits/stdc++.h>
using namespace std;
const int N=1000010;
int h[N],ne[N],e[N],idx;
int id[N];
int a[N];
int f[N];
bool st[N];
vector<int>v[N];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int find(int x){
    if(f[x]!=x){
        f[x]=find(f[x]);
    }
    return f[x];
}
int cnt=0;
int dp[N];
vector<int>c;
int main(){
    int n,m;
    cin>>n>>m;

    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++)cin>>a[i],f[i]=i,c.push_back(i);

    while(m--){
        int x,y;
        cin>>x>>y;
        int xx=find(x);
        int yy=find(y);
        if(a[x]==a[y]){
            f[xx]=yy;
        }else{
            if(a[y]>a[x]){
                swap(x,y);
            }
            add(y,x);
        }
        
    }
    sort(c.begin(),c.end(),[&](int x,int y){
        return a[x]<a[y];
    });
    memset(dp,-0x3f,sizeof dp);
    dp[find(1)]=1;
    for(auto u:c){
        for(int i=h[u];~i;i=ne[i]){
            int now=e[i];
            dp[find(now)]=max(dp[find(now)],dp[find(u)]+1);
        }
    }

    cout<<max(0,dp[find(n)])<<endl;
    
 }

解法2:类似差分约束的建图(数值小的向数值大的点连边,可以在后续dp处理中保证正确性),targan预处理缩点,判断是否成一个环,如果是一个环,可以得出环内相连的点均数值相等,然后向环外的点转移(dp),dp初始化时,n所在的块为1,其他为负无穷,答案是1所在的块

#include <iostream>
#include <cstring>
#include <algorithm>
#include<stack>
#include<vector>
using namespace std;
const int N=1000010;
int h[N],ne[N],e[N],idx;
int dfn[N],low[N];
int id[N];
int a[N];
vector<int>v[N];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int tmp=0;
stack<int>s;
int is_st[N];
int cnt=0;
void dfs(int u){
    dfn[u]=low[u]=++tmp;
    s.push(u);
    is_st[u]=1;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(!dfn[j]){
            dfs(j);
            low[u]=min(low[u],low[j]);
        }else if(is_st[j]){
            low[u]=min(low[u],dfn[j]);    
        }
    }
    if(dfn[u]==low[u]){
        cnt++;
        int y;
        while(1){
            y=s.top();
            s.pop();
            is_st[y]=false;
            id[y]=cnt;
            v[cnt].push_back(y);
            if(y==u)break;
        }
    }
}
int dp[N];
signed main(){
    int n,m;
    cin>>n>>m;
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++)cin>>a[i];
    while(m--){
        int x,y;
        cin>>x>>y;
        if(a[x]<=a[y])add(x,y);
        if(a[y]<=a[x])add(y,x);
    }
    for(int i=1;i<=n;i++){
        if(!dfn[i])dfs(i);
    }
    
    memset(dp,-0x3f,sizeof dp);
    dp[id[n]]=1;
    for(int i=1;i<=cnt;i++){
        for(auto c:v[i]){
            for(int j=h[c];~j;j=ne[j]){
                int now=e[j];
                if(id[c]!=id[now]){
                    dp[i]=max(dp[i],dp[id[now]]+1);
                }
            }
        }
    }
    cout<<max(0,dp[id[1]])<<endl;
    
 }

  • 11
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值