QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#96182#6307. Chase Game 2xuzhihaodedie#TL 20ms6376kbC++202.1kb2023-04-13 16:17:002023-04-13 16:17:04

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-04-13 16:17:04]
  • 评测
  • 测评结果:TL
  • 用时:20ms
  • 内存:6376kb
  • [2023-04-13 16:17:00]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define int long long
//#define x first
//#define y second
#define PII pair<int,int>
const int mod=998244353;
const int N=1e5+100;
vector<int>g[N];
int d[N],n,m;
vector<int>nums;
void dfs(int u,int fa)
{
	if(d[u]==1)	return;
	int t=0;
	for(int j : g[u])
	{
		if(j==fa)	continue;
		if(d[j]==1)	t++;
		dfs(j,u);
	}
	//if(t>0) q.push(t);
	if(t>0) nums.push_back(t);
}
bool check(int x)
{
	int l=n+1,r=0;
	int sum=0;
	for(int i=1;i<=m;i++)
	{
		sum+=nums[i];
		if(sum>x)
		{
			l=i;
			break;
		}		
	}
	sum=0;
	for(int i=m;i>=1;i--)
	{
		sum+=nums[i];
		if(sum>x)
		{
			r=i;
			break;
		}
	}
	return l>r;
}
void solve(){
	scanf("%lld",&n);
	int res=0,ans=0;
//	for(int i=1;i<=n;i++) {
//		int x;
//		cin>>x;
//		nums.push_back(x);
//		res+=x;
//	}
	for(int i=1;i<=n;i++)	g[i].clear(),d[i]=0;
	nums.clear();
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%lld%lld",&u,&v);
		g[u].push_back(v),g[v].push_back(u);
		d[u]++,d[v]++;
	}
	for(int i=1;i<=n;i++)
		if(d[i]==1)	res++;
	if(res>=n-1)
	{
		puts("-1");
		return;
	}
	for(int i=1;i<=n;i++)
		if(d[i]>1)
		{
			dfs(i,-1);
			break;
		}
	m=nums.size();
	//nums.push_back(0);
	sort(nums.begin(),nums.end());
	while(nums.size()>1) {
		int sum=0;
		int pos;
		for(int i=0;i<m;i++) {
			sum+=nums[i];
			if(sum>res-sum) {
				pos=i-1;
				sum-=nums[i];
				break;
			}
		}
		//cout<<sum<<endl;
		vector<int> v;
		//for(int i=0;i<pos;i++) nums.pop_front();
		int ret=res-2*sum;
		//cout<<ret<<endl;
		for(int i=0;i<m-1-pos;i++) {
			v.push_back(ret/(m-pos-1));
		}
		int req=ret%(m-pos-1);
		for(int i=v.size()-1;i>=v.size()-req;i--) {
			v[i]++;
		}
		res=0;
		for(auto it:v) res+=it;
		//cout<<endl;
		nums=v;
		ans+=sum;
	}
	if(nums.size()) ans+=nums[0];
	cout<<ans<<endl;
}
signed main() {
	int T=1;
//	freopen("C:\\Users\\86153\\Desktop\\data.in.txt","r",stdin); //´ÓÎļþÖлñÈ¡ÊäÈëÁ÷
//	freopen("C:\\Users\\86153\\Desktop\\data1.out.txt","w",stdout); //°ÑÊä³öתµ½ÎļþÖÐ
	//init();
	cin>>T;
	while(T--) {
		solve();
	}
	//return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 3ms
memory: 6376kb

input:

4
2
1 2
4
1 2
2 3
3 4
4
1 2
2 3
2 4
5
1 2
2 3
3 4
3 5

output:

-1
1
-1
2

result:

ok 4 number(s): "-1 1 -1 2"

Test #2:

score: 0
Accepted
time: 20ms
memory: 5776kb

input:

10000
4
1 2
1 3
3 4
4
1 2
1 3
1 4
4
1 2
2 3
1 4
5
1 2
2 3
1 4
4 5
5
1 2
2 3
3 4
4 5
4
1 2
2 3
2 4
5
1 2
1 3
2 4
2 5
4
1 2
2 3
1 4
5
1 2
1 3
2 4
1 5
5
1 2
2 3
3 4
2 5
5
1 2
1 3
2 4
2 5
4
1 2
1 3
3 4
5
1 2
1 3
3 4
1 5
4
1 2
1 3
1 4
5
1 2
1 3
3 4
3 5
5
1 2
2 3
3 4
3 5
4
1 2
1 3
2 4
5
1 2
2 3
2 4
3 5
5
...

output:

1
-1
1
1
1
-1
2
1
2
2
2
1
2
-1
2
2
1
2
2
1
1
1
-1
2
2
2
1
-1
1
1
2
1
1
-1
1
2
1
1
1
-1
1
1
2
2
2
1
1
1
-1
1
2
1
1
2
1
2
1
1
2
-1
-1
-1
2
2
2
1
1
1
2
2
2
-1
1
2
-1
1
1
-1
2
-1
-1
1
2
2
2
1
1
1
1
1
1
1
1
1
2
-1
1
1
2
-1
2
1
1
1
-1
2
-1
1
-1
-1
2
-1
2
1
2
2
1
1
1
1
2
1
1
1
1
-1
2
1
2
1
1
1
1
1
1
1
2
-1...

result:

ok 10000 numbers

Test #3:

score: -100
Time Limit Exceeded

input:

10000
9
1 2
2 3
3 4
4 5
1 6
6 7
5 8
7 9
9
1 2
2 3
2 4
1 5
2 6
4 7
6 8
1 9
9
1 2
2 3
1 4
4 5
5 6
4 7
2 8
1 9
10
1 2
2 3
1 4
3 5
3 6
2 7
6 8
6 9
6 10
10
1 2
1 3
3 4
2 5
1 6
5 7
4 8
2 9
7 10
10
1 2
2 3
2 4
1 5
3 6
6 7
5 8
4 9
9 10
9
1 2
2 3
2 4
1 5
3 6
2 7
1 8
2 9
9
1 2
1 3
2 4
1 5
3 6
3 7
7 8
8 9
10
1...

output:

1

result: