Znzryb's Blog

2025牛客暑期多校训练营1—K.Museum Acceptance（分层图，把一个点分为多个点）

Posted on 2025-07-15 Edited on 2026-04-06

思路讲解

非常暴力的模拟，唯一记忆的地方就在于这个划线。

FOR(i,1,N){
	if(ans[i][1]!=0){
		cout<<ans[i][1]<<"\n";
		continue;
	}
	ll cur=i,id=1;
	vector<arr2> wp;
	while(true){
		wp.pb({cur,id});
		ll v=g[cur][id];
		FOR(i,1,num[v]){
			if(g[v][i]==cur){
				id=i;
			}
		}
		if(id+1<=num[v]){
			id+=1;
		}else{
			id=1;
		}
		if(ans[v][id]){
			ll lans=ans[v][id]+SZ(st);
			if(ans[v][id]==-1) lans=SZ(st);
			for(auto &[x,idx]:wp){
				ans[x][idx]=lans;
			}
			break;
		}
		ans[v][id]=-1;
		arr2 t={cur,v};sort(all(t));
		st.insert(t);
		cur=v;
	}
	cout<<ans[i][1]<<"\n";
	st.clear();
}

AC代码

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78205559

源代码

// Problem: Museum Acceptance
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/108298/K
// Memory Limit: 1024 MB
// Time Limit: 4000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(1e6)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e6+10;constexpr ll MAXV=(ll)1e6+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT,A[MAXN];
/*

*/
ll num[MAXN];
ll g[MAXN][5];
ll fa[MAXN];
ll ans[MAXN][5];
ll find(ll x){
	if(fa[x]==x) return fa[x];
	return fa[x]=find(fa[x]);
}
void merge(ll a,ll b){
	ll pa=find(a),pb=find(b);
	fa[pa]=pb;
}
inline void __(){
	cin>>N;
	FOR(i,1,N){
		fa[i]=i;
	}
	FOR(i,1,N){
		cin>>num[i];
		FOR(j,1,num[i]){
			cin>>g[i][j];
		}
	}
	// unordered_map<pll,bool> st;
	// unordered_map<pll,bool> s;
	// FOR(i,1,N){
		// FOR(j,1,num[i]){
			// s[(pll){i,j}]=false;
		// }
	// }
	set<arr2> st;
#ifdef LOCAL
    cerr << "OK1" << '\n';
#endif
	FOR(i,1,N){
		if(ans[i][1]!=0){
			cout<<ans[i][1]<<"\n";
			continue;
		}
		ll cur=i,id=1;
		vector<arr2> wp;
		while(true){
			wp.pb({cur,id});
			ll v=g[cur][id];
			FOR(i,1,num[v]){
				if(g[v][i]==cur){
					id=i;
				}
			}
			if(id+1<=num[v]){
				id+=1;
			}else{
				id=1;
			}
			if(ans[v][id]){
				ll lans=ans[v][id]+SZ(st);
				if(ans[v][id]==-1) lans=SZ(st);
				for(auto &[x,idx]:wp){
					ans[x][idx]=lans;
				}
				break;
			}
			ans[v][id]=-1;
			arr2 t={cur,v};sort(all(t));
			st.insert(t);
			cur=v;
		}
		cout<<ans[i][1]<<"\n";
		st.clear();
	}
#ifdef LOCAL
    FOR(i,1,N){
		FOR(j,1,3){
			cerr<<ans[i][j]<<" ";
		}
		cerr<<"\n";
	}
#endif
	
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	// cin>>lT;
	// while(lT--)
		__();
	return 0;
}
/*

*/

心路历程（WA，TLE，MLE……）

START194 - Good Arrays (Not Cyclic)（线性dp）

Posted on 2025-07-15 Edited on 2026-04-06

思路讲解

这个的状态定义比较巧妙，定义的是a1和aN/2的差，a1+a2与aN/2+1+aN/2+2的差

FOR(i,1,N/2){
	FOR(d,-(N-1),N-1){
		FOR(j,0,N/2*N/2){
			if(j+d<0) continue;
			if(d<=0){
				dp[i][j]+=dp[i-1][j+d]*ct_[i][abs(d)];
				dp[i][j]%=mod;
			}else{
				dp[i][j]+=dp[i-1][j+d]*ct[i][d];
				dp[i][j]%=mod;
			}
			
		}
	}
}

AC代码

https://www.codechef.com/viewsolution/1172894021

源代码

// Problem: Good Arrays (Not Cyclic)
// Contest: CodeChef - START194B
// URL: https://www.codechef.com/START194B/problems/CNTGOODARRH
// Memory Limit: 256 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(1e2)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e4+10;constexpr ll MAXV=(ll)1e6+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT;
char S[MAXN][MAXN];
//   dp[i][j] 第i行，diff=j。
ll dp[MAXN][MAXM];
// 辅助数组（count 简写）
// ct[i][j] i row, diff=j
ll ct[MAXN][MAXN];
// ct_[i][j] i row, diff=-j
ll ct_[MAXN][MAXN];
/*

*/
void init(){
	FOR(i,0,N+5){
		FOR(j,0,N+5){
			ct[i][j]=0;
			ct_[i][j]=0;
		}
	}
	FOR(i,0,N+5){
		FOR(j,0,N/2*N/2+5){
			dp[i][j]=0;
		}
	}
	dp[0][0]=1;
}
inline void solve(){
	cin>>N;
	init();
	vector<ll> cnt(N+5,0);
	FOR(i,1,N){
		FOR(j,1,N){
			cin>>S[i][j];
			if(S[i][j]=='1'){
				cnt[i]++;
			}
		}
	}
	ll ans=1;	// 统计总数目
	FOR(i,1,N){
		ans*=cnt[i];
		ans%=mod;
	}
	// 一般来说，求什么，dp里装的就是什么。
	// 那么这里面dp装的就是有多少个坏数组。
	// 那么我们一层一层推怎么样？
	ll n=N/2;
	FOR(i,1,N/2){	// 辅助数组计算
		FOR(d,0,N-1){
			FOR(j,1,N){
				if(j+d>N) break;
				if(S[i][j]=='1' && S[i+n][j+d]=='1'){
					ct[i][d]++;
					ct[i][d]%=mod;
				}
			}
		}
	}
	FOR(i,1,N/2){	// 辅助数组计算
		FOR(d,0,N-1){
			FOR(j,d+1,N){
				if(S[i][j]=='1' && S[i+n][j-d]=='1'){
					ct_[i][d]++;
					ct_[i][d]%=mod;
				}
			}
		}
	}
#ifdef LOCAL
    FOR(i,1,n){
    	FOR(j,0,N-1){
    		cerr<<ct[i][j]<<" ";
    	}
    	cerr<<"\n";
    }
    FOR(i,1,n){
    	FOR(j,0,N-1){
    		cerr<<ct_[i][j]<<" ";
    	}
    	cerr<<"\n";
    }
#endif
	FOR(i,1,N/2){
		FOR(d,-(N-1),N-1){
			FOR(j,0,N/2*N/2){
				if(j+d<0) continue;
				if(d<=0){
					dp[i][j]+=dp[i-1][j+d]*ct_[i][abs(d)];
					dp[i][j]%=mod;
				}else{
					dp[i][j]+=dp[i-1][j+d]*ct[i][d];
					dp[i][j]%=mod;
				}
				
			}
		}
	}
#ifdef LOCAL
    FOR(i,1,n){
    	FOR(j,0,N/2*N/2){
    		cerr<<dp[i][j]<<" ";
    	}
    	cerr<<"\n";
    }
#endif
	ans-=dp[n][0];
	ans%=mod;
	if(ans<0) ans+=mod;
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	cin>>lT;
	while(lT--)
		solve();
	return 0;
}
/*
AC
https://www.codechef.com/viewsolution/1172894021
*/

心路历程（WA，TLE，MLE……）

P3806 【模板】点分治 1

Posted on 2025-07-14 Edited on 2026-04-06

思路讲解

AC代码

源代码

心路历程（WA，TLE，MLE……）

牛客练习赛142-D.第四次忍界大战（会被误认为树上dp）（先计算，后合并）（点分治）

Posted on 2025-07-12 Edited on 2026-04-06

思路讲解

那么根据我的经验来说，如果说这个东西要求连续性（如距离，字母排列等等），大概率是树上dp，像这种具有超距作用的，那么大概率就是要配合DSU了。

核心代码如下，注意，先计算，后合并。

FOR(i,1,N){		// 特判两个点的情况
	for(auto &v:g[i]){
		if(A[v]==A[i]) ++ans;
	}
}
// 生成的这张图相当于其权值都是大于等于key的（至少是相连的图，孤立点我们不知道）
// 这个时候就是要看两个相同的A值点，是不是连接于这个图上面。
// 那么我们发现不用把问题想的非常复杂，只需要考虑key-1（即后面一个k）就行了
// 最重要的是遵循"先计数，后合并"的原则
for(auto it=mp.rbegin();it!=mp.rend();it++){
	// auto& 相当于定义了一个别名
	auto& ls=it->second;ll key=it->first;
	map<ll,ll> cnt;
	// 先计数
	for(auto &u:ls){
		vector<ll> wp;
		for(auto &v:g[u]){
			if(A[v]<=key) continue; 
			wp.pb(find(v));
		}
		wp.resize(unique(all(wp))-wp.begin());
		for(auto &v:wp){
			++cnt[find(v)];
		}
	}
	for(auto &t:cnt){
		ll n=t.second;
		ans+=(n-1+1)*(n-1);	// 2Cn * 2 在n个中选两个，然后由于先选哪个比较重要*2
	}
	// 后合并
	for(auto &u:ls){
		for(auto &v:g[u]){
			if(A[v]<key) continue;
			merge(u,v);
		}
	}
}

那么这种复杂路径问题，点分治也能解决（应该来说更泛用）。

AC代码

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78161052

源代码（并查集）

// Problem: 第四次忍界大战
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/112575/D
// Memory Limit: 512 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(1e5)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e6+10;constexpr ll MAXV=(ll)1e6+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT,A[MAXN];
/*

*/

map<ll, vector<ll > > mp;
bool cmp(const arr2 &a,const arr2 &b){
	return min(a[0],a[1])<min(b[0],b[1]);
}
ll fa[MAXN];
ll find(ll x){
	if(x==fa[x]) return fa[x];
	return fa[x]=find(fa[x]);
}
void merge(ll a,ll b){
	fa[find(a)]=find(b);
}
vector<ll> g[MAXN];		// 记得要清空
inline void solve(){
	cin>>N;
	FOR(i,1,N){
		fa[i]=i;
	}
	FOR(i,1,N){
		cin>>A[i];
		mp[A[i]].pb(i);
	}
	// map<ll, vector<arr2> > E;
	FOR(i,1,N-1){
		ll u,v;
		cin>>u>>v;
		// E[min(A[u],A[v])].pb({u,v});
		g[u].pb(v);g[v].pb(u);
	}
	ll ans=0;
	FOR(i,1,N){		// 特判两个点的情况
		for(auto &v:g[i]){
			if(A[v]==A[i]) ++ans;
		}
	}
	// 生成的这张图相当于其权值都是大于等于key的（至少是相连的图，孤立点我们不知道）
	// 这个时候就是要看两个相同的A值点，是不是连接于这个图上面。
	// 那么我们发现不用把问题想的非常复杂，只需要考虑key-1（即后面一个k）就行了
	// 最重要的是遵循"先计数，后合并"的原则
	for(auto it=mp.rbegin();it!=mp.rend();it++){
		// auto& 相当于定义了一个别名
		auto& ls=it->second;ll key=it->first;
		map<ll,ll> cnt;
		// 先计数
		for(auto &u:ls){
			vector<ll> wp;
			for(auto &v:g[u]){
				if(A[v]<=key) continue; 
				wp.pb(find(v));
			}
			wp.resize(unique(all(wp))-wp.begin());
			for(auto &v:wp){
				++cnt[find(v)];
			}
		}
		for(auto &t:cnt){
			ll n=t.second;
			ans+=(n-1+1)*(n-1);	// 2Cn * 2 在n个中选两个，然后由于先选哪个比较重要*2
		}
		// 后合并
		for(auto &u:ls){
			for(auto &v:g[u]){
				if(A[v]<key) continue;
				merge(u,v);
			}
		}
	}
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	// cin>>lT;
	// while(lT--)
		solve();
	return 0;
}
/*
96% https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78160882
AC https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78161052
*/

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78175127

源代码（点分治）

// Problem: 第四次忍界大战
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/112575/D
// Memory Limit: 512 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(1e6)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e6+10;constexpr ll MAXV=(ll)1e5+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT,A[MAXN];
/*

*/
inline void solve(){
	cin>>N;
	vector<vector<ll> > g(N+5);
	FOR(i,1,N){
		cin>>A[i];
	}
	FOR(i,1,N-1){
		ll u,v;
		cin>>u>>v;
		g[u].pb(v);g[v].pb(u);
	}
	vector<ll> sz(N+5,0);ll mxTree=INF;
	vector<bool> vis(N+5,0);ll root=0;
	ll ans=0;
	// FOR(i,1,N){		// 特判两个点的情况
		// for(auto &v:g[i]){
			// if(A[v]==A[i]) ++ans;
		// }
	// }
	// ans/=2;
// #ifdef LOCAL
    // debug(ans);
// #endif
	auto get=[&](auto &&self,ll u,ll p,ll tot)->void{
		ll val=0;sz[u]=1;
		for(auto &v:g[u]){
			if(v==p || vis[v]) continue;
			self(self,v,u,tot);
			sz[u]+=sz[v];
			val=max(val,sz[v]);
		}
		val=max(val,tot-sz[u]);
		if(val<mxTree){
			mxTree=val;
			root=u;
		}
	};
	auto cal=[&](ll u)->void{
		map<ll,ll> cnt;vector<ll> wp;
		// 那么众所周知，我们只考虑经过u的路径（点分治）
		auto dfs=[&](auto &&self,ll u,ll p,ll mn)->void{
			if(mn>A[u]){
				// if(cnt.count(A[u])){
					// ans+=cnt[A[u]];
				// }
				// ++cnt[A[u]];
				wp.pb(A[u]);
			}
			mn=min(A[u],mn);
			for(auto &v:g[u]){
				if(v==p || vis[v]) continue;
				self(self,v,u,mn);
			}
		};
#ifdef LOCAL
	debug(root);
   	debug(ans);
#endif
		// 保证计入答案的路径是经过u点的
		for(auto &v:g[u]){
			if(vis[v]) continue;	// 你不能就不判断了
			dfs(dfs,v,u,INF);
// #ifdef LOCAL
    // if(root==1){
    	// for(auto &a:wp){
    		// debug(a);
    	// }
    	// debug(ans);
    // }
// #endif
			for(auto &a:wp){
				// 跨区域合并不行，因为根节点值太低了
				if(a>=A[u]) continue;
				if(cnt.count(a)){
					ans+=cnt[a];
				}
			}
			for(auto &a:wp){
				++cnt[a];
			}

			wp.clear();
		}
		// 根节点直接和子树上的点匹配
		ans+=cnt[A[u]];
#ifdef LOCAL
    debug(ans);
#endif
	};
	// 主程序
	auto dfz=[&](auto &&self,ll u)->void{
		vis[u]=true;	// u一定是一个重心
		cal(u);
		for(auto &v:g[u]){
			if(vis[v]) continue;
			mxTree=INF;
			get(get,v,u,sz[v]);
// #ifdef LOCAL
    // debug(root);
// #endif
			self(self,root);
		}
	};
	get(get,1,1,N);
#ifdef LOCAL
    debug(root);
#endif
	dfz(dfz,root);
	ans*=2;		// 该路径是考虑顺序的
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	// cin>>lT;
	// while(lT--)
		solve();
	return 0;
}
/*
WA 60% https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78164958
hack:
10
4 4 5 3 2 5 4 4 5 5
5 10
1 3
5 6
4 6
9 2
1 7
8 9
2 3
10 7

ans:
6
AC https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78175127
*/

心路历程（WA，TLE，MLE……）

WA了的点分治代码，过了60%，我认为错误就在这个我的这个点分治统计答案他有点问题，统计到的不一定都经过u，这是违背点分治思想的。

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78164958

源代码（点分治）

// Problem: 第四次忍界大战
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/112575/D
// Memory Limit: 512 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(1e6)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e6+10;constexpr ll MAXV=(ll)1e5+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT,A[MAXN];
/*

*/
inline void solve(){
	cin>>N;
	vector<vector<ll> > g(N+5);
	FOR(i,1,N){
		cin>>A[i];
	}
	FOR(i,1,N-1){
		ll u,v;
		cin>>u>>v;
		g[u].pb(v);g[v].pb(u);
	}
	vector<ll> sz(N+5,0);ll mxTree=INF;
	vector<bool> vis(N+5,0);ll root=0;
	ll ans=0;
	FOR(i,1,N){		// 特判两个点的情况
		for(auto &v:g[i]){
			if(A[v]==A[i]) ++ans;
		}
	}
	ans/=2;
	auto get=[&](auto &&self,ll u,ll p,ll tot)->void{
		ll val=0;sz[u]=1;
		for(auto &v:g[u]){
			if(v==p || vis[v]) continue;
			self(self,v,u,tot);
			sz[u]+=sz[v];
			val=max(val,sz[v]);
		}
		val=max(val,tot-sz[u]);
		if(val<mxTree){
			mxTree=val;
			root=u;
		}
	};
	auto cal=[&](ll u)->void{
		map<ll,ll> cnt;
		// 那么众所周知，我们只考虑经过u的路径（点分治）
		auto dfs=[&](auto &&self,ll u,ll p,ll mn)->void{
			if(mn>A[u]){
				if(cnt.count(A[u])){
					ans+=cnt[A[u]];
				}
				++cnt[A[u]];
			}
			mn=min(A[u],mn);
			for(auto &v:g[u]){
				if(v==p || vis[v]) continue;
				self(self,v,u,mn);
			}
		};
		dfs(dfs,u,u,INF);
	};
	// 主程序
	auto dfz=[&](auto &&self,ll u)->void{
		vis[u]=true;	// u一定是一个重心
		cal(u);
		for(auto &v:g[u]){
			if(vis[v]) continue;
			mxTree=INF;
			get(get,v,v,sz[v]);
			self(self,root);
		}
	};
	get(get,1,1,N);
	dfz(dfz,root);
	ans*=2;		// 该路径是考虑顺序的
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	// cin>>lT;
	// while(lT--)
		solve();
	return 0;
}
/*

*/

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78165041

然后改了统计代码以后WA的更多了

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78165092

AI指出了一个不对的地方，现在正确率提升到了84%

// 保证计入答案的路径是经过u点的
for(auto &v:g[u]){
	if(vis[v]) continue;	// 你不能就不判断了
	dfs(dfs,v,u,A[u]);
	for(auto &a:wp){
		if(cnt.count(a)){
			ans+=cnt[a];
		}
	}
	for(auto &a:wp){
		++cnt[a];
	}
	wp.clear();
}

牛客练习赛142-C.战前准备

Posted on 2025-07-12 Edited on 2026-04-06

思路讲解

差分做法，只要用差分把所有坏区间都标记出来，剩下的就是不可行的。

FOR(i,0,M-2){
	for(auto &p:pos[i+1]){
		auto it=lower_bound(all(pos[i]),p);
		if(it==pos[i].begin()){
			diff[1]+=1;
			// p+1是可行的（至少从这个p看来）
			diff[p+1]-=1;
		}else{
			it--;ll idx=*it;
			diff[idx+1]+=1;
			// p+1是可行的（至少从这个p看来）
			diff[p+1]-=1;
		}
	}
}

AC代码

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78135312

源代码

// Problem: 战前准备
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/112575/C
// Memory Limit: 512 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

using  ll = long long;using ull = unsigned long long;
using DB=double;using LD=long double;
// using i128 = __int128;
using pdd = pair<DB,DB>;using plb = pair<ll,bool>;
using pll = pair<ll,ll>;
using arr3 = array<ll,3> ;using arr2 = array<ll,2>;
constexpr ll MAXN=static_cast<ll>(2e5)+10,INF=static_cast<ll>(1e17)+9; // 1e18+9也是素数
constexpr ll MAXM=(ll)1e6+10;constexpr ll MAXV=(ll)1e6+10;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;const double pi=acos(-1.0);

ll N,M,Q,X,K,T,lT,A[MAXN];
/*

*/
vector<ll> pos[MAXN];
ll diff[MAXN],Sum[MAXN];
inline void init(){

}
inline void solve(){
	cin>>N>>M;
	init();
	FOR(i,1,N){
		cin>>A[i];
		pos[A[i]].pb(i);
	}
	if(M==1){
		cout<<N<<"\n";return;
	}
	FOR(i,0,M-2){
		ll las=0;
		for(auto &p:pos[i+1]){
			auto it=lower_bound(all(pos[i]),p);
			if(it==pos[i].begin()){
				diff[1]+=1;
				// p+1是可行的（至少从这个p看来）
				diff[p+1]-=1;
			}else{
				it--;ll idx=*it;
				diff[idx+1]+=1;
				// p+1是可行的（至少从这个p看来）
				diff[p+1]-=1;
			}
		}
	}
	FOR(i,1,N){
		Sum[i]=Sum[i-1]+diff[i];
	}
	ll ans=0;
	FOR(i,1,N){
		if(Sum[i]==0){
			++ans;
		}
	}
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	// cin>>lT;
	// while(lT--)
		solve();
	return 0;
}
/*
AC
https://ac.nowcoder.com/acm/contest/view-submission?submissionId=78135312
*/