kmjp's blog

競技プログラミング参加記です

yukicoder : No.1364 [Renaming] Road to Cherry from Zelkova

だいぶ記事未作成の問題が溜まってきた。
https://yukicoder.me/problems/no/1364

問題

距離付きの有向辺からなるグラフが与えられる。
0番の頂点からN番の頂点に至る全経路の距離の総和を求めよ。

解法

0番の頂点から到達可能で、N番の頂点に到達可能な、2要素以上の閉路があった場合、その閉路で距離を無限に稼げる。
それ以外の場合は、頂点をトポロジカルソートして、到達する経路の数と距離の総和をDPしていこう。

int N,M;
vector<vector<ll>> E[202020];
vector<int> RE[202020];
int reach[202020];
class SCC {
public:
	static const int MV = 2025000;
	vector<vector<int> > SC; int NV,GR[MV];
private:
	vector<int> E[MV], RE[MV], NUM; int vis[MV];
public:
	void init(int NV) { this->NV=NV; for(int i=0;i<NV;i++) { E[i].clear(); RE[i].clear();}}
	void add_edge(int x,int y) { E[x].push_back(y); RE[y].push_back(x); }
	void dfs(int cu) { vis[cu]=1; for(int i=0;i<E[cu].size();i++) if(!vis[E[cu][i]]) dfs(E[cu][i]); NUM.push_back(cu); }
	void revdfs(int cu, int ind) { int i; vis[cu]=1; GR[cu]=ind; SC[ind].push_back(cu);
		FOR(i,RE[cu].size()) if(!vis[RE[cu][i]]) revdfs(RE[cu][i],ind);}
	void scc() {
		int c=0,i; SC.clear(); SC.resize(NV); NUM.clear();
		assert(NV);
		FOR(i,NV) vis[i]=0; FOR(i,NV) if(!vis[i]) dfs(i); FOR(i,NV) vis[i]=0;
		for(int i=NUM.size()-1;i>=0;i--) if(!vis[NUM[i]]){
			SC[c].clear(); revdfs(NUM[i],c); sort(SC[c].begin(),SC[c].end()); c++;
		}
		SC.resize(c);
	}
};
SCC scc;

ll num[202020];
ll sum[202020];
int in[202020];
const ll mo=1000000007;
void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N>>M;
	
	scc.init(N+1);
	FOR(i,M) {
		cin>>x>>y>>j>>l;
		E[x].push_back({y,j,l});
		RE[y].push_back(x);
		scc.add_edge(x,y);
	}
	queue<int> Q;
	reach[N]=1;
	Q.push(N);
	while(Q.size()) {
		x=Q.front();
		Q.pop();
		FORR(e,RE[x]) if(reach[e]==0) {
			reach[e]=1;
			Q.push(e);
		}
	}
	scc.scc();
	if(reach[0]==0) return _P("0\n");
	FOR(i,N) if(reach[i]&&scc.SC[scc.GR[i]].size()>1) {
		cout<<"INF"<<endl;
		return;
	}
	
	FOR(i,N+1) {
		FORR(e,E[i]) if(reach[i]&&reach[e[0]]) in[e[0]]++;
	}
	
	FOR(i,N+1) if(reach[i]&&in[i]==0) Q.push(i);
	num[0]=1;
	sum[0]=0;
	
	while(Q.size()) {
		x=Q.front();
		Q.pop();
		FORR(e,E[x]) if(reach[e[0]]) {
			(num[e[0]]+=num[x]*e[2])%=mo;
			(sum[e[0]]+=e[2]*(sum[x]+num[x]*e[1]%mo))%=mo;
			in[e[0]]--;
			if(in[e[0]]==0) Q.push(e[0]);
		}
	}
	
	cout<<sum[N]<<endl;
}

まとめ

こちらは割とすんなり。