Codeforces Round 787 (Div. 3)D. Vertical Paths

2023-12-14 17:47:54

题目链接
题意:给定一棵树,将这棵树划分成几天互不相交的链,要求最小化链的数量
思路:每个叶子节点一定在一条链中,所以链的数量就是叶子节点的数量,从叶子节点往上跳直到根节点,边跳边标记,路径上所有点都属于这条链。
坑:

  1. 数据大时,不要轻易使用memset不然会t到起飞
  2. vector不要开太多就比如不要vector<int>a[N]这样也会t
  3. 多组测试样例一定要清数组
#include <bits/stdc++.h> 
#define rep(i,a,b) for(register int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(register int i = (a); i >= (b); --i)
#define ls p<<1
#define rs p<<1|1
#define PII pair<int, int>
#define ll long long
#define ull unsigned long long
#define db double
#define endl '\n'
#define debug(a) cout<<#a<<"="<<a<<endl;
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define INF 0x3f3f3f3f

using namespace std;
const int N = 2e5+10;
int t; 
int d[N], p[N], vis[N], n;
void solve()
{
	cin >> n;
	rep(i,1,n)
	{
		int x; cin >> x;
		p[i] = x;
		d[x]++;
	}
	if(n==1)
	{
	    cout << 1 << endl << 1 << endl << 1 << endl;
	    cout << endl;
	    return;
	}
	int ans = 0;
	vector<int>a[n];
	rep(i,1,n)
		if(!d[i])
		{
			int k = i;
			while(!vis[k] && k != p[k])
			{
				a[ans].push_back(k);
				vis[k] = 1;
				k = p[k];
			}	
			if(k == p[k] && !vis[k])
			{
				a[ans].push_back(k);
				vis[k] = 1;
			}		
			ans++;
		}
	rep(i,1,n)	d[i] = 0, vis[i] = 0;
	cout << ans << endl;
	rep(i,0,ans-1)
	{
		cout << a[i].size() << endl;
		fep(i,a[i].size()-1,0)	cout << x << ' ';
		cout << endl;
	}
	cout << endl;
}

int main()
{
	IOS	
  	freopen("1.in", "r", stdin);
	cin >> t;
	while(t --)	
	solve();
	return 0;
}

文章来源:https://blog.csdn.net/weixin_61426225/article/details/134995602
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。