食物链 bzoj-4562 Haoi-2016
题目大意:给你n个点,m条边的DAG,求所有的满足条件的链,使得每条链的起点是一个入度为0的点,中点是一条出度为0的点。
注释:$1\le n\le 10^5$,$1\le m\le 2*10^5$。
想法:考试T2,全场切
动态规划
状态:dp[i]表示从这个点到出度为0的点的方案数。
转移:dp[i]+=dp[to[i]]
然后用记忆化爆搜即可。
最后,附上丑陋的代码... ...
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#define MOD 1000000007
#define N 100010
#define M 200010
using namespace std;
typedef long long ll;
ll dp[N];
int num[N];
char s[N][100];
map<string,int>mp;
// ll stack[M<<1];
// ll val(int a)
// {
// ll ans1=1,ans2=1,ans3=1;
// int k=strlen(s[a]+1);
// for(int i=1;i<=k/3;i++) ans1=ans1*(s[i]-'0')%mod;
// for(int i=k/3+1;i<=k/3*2;i++) ans2=ans2*(s[i]-'0')%mod;
// for(int i=k/3*2+1;i<=k;i++) ans3=ans3*(s[i]-'0')%mod;
// }
struct Node
{int val,id,now;
}qqq[N];
// int map[1000000];
int head[N],to[M],nxt[M],tot;
inline void add(int x,int y)
{to[++tot]=y;nxt[tot]=head[x];head[x]=tot;
}
int dfs(int pos,int fa)
{if(dp[pos]) return dp[pos];for(int i=head[pos];i;i=nxt[i]){if(to[i]==fa) continue;dp[pos]+=dfs(to[i],pos);dp[pos]%=MOD;}if(dp[pos]==0) dp[pos]=1;return dp[pos]%MOD;
}
inline bool cmp1(Node o,Node oo)
{return o.val<oo.val;
}
inline bool cmp2(Node o,Node oo)
{return o.id<oo.id;
}
ll sum[N];
int main()
{// freopen("chain.in","r",stdin);// freopen("chain.out","w",stdout);ll ans=0,/* top=0, */cnt=0;int m;scanf("%*d%d",&m);// puts("Fuck");for(int i=1;i<=m;i++){string s1,s2;cin >> s1 >> s2 ;if(!mp[s1]) mp[s1]=++cnt;if(!mp[s2]) mp[s2]=++cnt;add(mp[s1],mp[s2]);num[mp[s1]]++;sum[mp[s2]]++;// scanf("%s%s",s[1]+1,s[2]+1);// int x=val(1),y=val(2);// printf("Bitch %d %d\n",x,y);// top+=2;// stack[++top]=x,stack[++top]=y;// qqq[top-1].val=x,qqq[top].val=y;// qqq[top-1].id=top-1,qqq[top].id=top;// add(x,y);}// sort(qqq+1,qqq+top+1,cmp1);// qqq[0].val=qqq[1].val-1;// for(int i=1;i<=top;i++)// {// if(qqq[i-1].val!=qqq[i].val) cnt++;// qqq[i].now=cnt;// }// for(int i=1;i<=top;i++)// {// printf("Woc %d %d\n",qqq[i].val,qqq[i].now);// }// sort(qqq+1,qqq+top+1,cmp2);// for(int i=1;i<=top;i++)// {// printf("Fuck %d\n",qqq[i].now);// }// for(int i=2;i<=top;i+=2)// {// add(qqq[i-1].now,qqq[i].now);// sum[qqq[i].now]++;// maxnow=max(maxnow,sum[qqq[i].now]);// maxnow=max(maxno)// }for(int i=1;i<=cnt;i++){if(!sum[i]&&!num[i]) continue;if(!sum[i]) ans+=dfs(i,0);// printf("CaoNiMa %d\n",i);}// for(int i=1;i<=cnt;i++)// {// printf("EYWR %lld\n",dp[i]);// }printf("%lld\n",ans);return 0;
}
小结:先想到dp是关键。学长写的全是什么toposort..都是O(n)的...