Codeforces Round #547 (Div. 3) EF1F2G 题解

E. Superhero Battle

题意：设sum[i]为di数组的前缀和，求一个最小的 i ，使得 sum[i]+i/n*sum[n]+H<=0。

水题。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=2e5+1;
ll a[maxn],sum[maxn],H,ans=1e18;
int main()
{
	int n;
	scanf("%lld%d",&H,&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		sum[i]=sum[i-1]+a[i];
		if(sum[i]+H<=0)
		ans=min(ans,1ll*i);
	}
	if(sum[n]>=0&&ans>n)
	{
		puts("-1");
		return 0;
	}
	else if(ans<=n)
	{
		printf("%lld\n",ans);
		return 0;
	}
	for(int i=1;i<=n;i++)
	{
		ll t=H+sum[i];
		ll res=t/(-sum[n]);
		if(t%(-sum[n]))res++;
		res=res*n+i;
		ans=min(ans,res);
	}
	printf("%lld\n",ans);
}

F2. Same Sum Blocks (Hard)

题意：选择尽可能多不相交的区间，使得这些区间的和相等。

思路：n最大1500，最多的区间才不超过2e6，因此把所有区间和等于 k 的区间存到一个数组，枚举该数组最多有多少个不相交区间即可。

复杂度n*nlogn。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1501;
int a[maxn],b[maxn*maxn],sum[maxn],sz;
struct node
{
	int l,r;
	node(int a,int b)
	{
		l=a,r=b;
	}
};
vector<node>G[maxn*maxn/2];
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		sum[i]=sum[i-1]+a[i];
		int res=0;
		for(int j=i;j;j--)
		{
			res+=a[j];
			b[++sz]=res;
		}
	}
	sort(b+1,b+1+sz);
	sz=unique(b+1,b+1+sz)-b-1;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=i;j++)
	{
		int res=sum[i]-sum[j-1];
		int k=lower_bound(b+1,b+1+sz,res)-b;
		G[k].push_back(node(j,i));
	}
	int ans=0,cur;
	for(int i=1;i<=sz;i++)
	{
		int res=0,p=0;
		for(int j=0;j<G[i].size();j++)
		if(G[i][j].l>p)
		{
			res++;
			p=G[i][j].r; 
		}
		if(res>ans)
		{
			ans=res,cur=i;
		}
	}
	printf("%d\n",ans);
	int p=0;
	for(int i=0;i<G[cur].size();i++)
	if(G[cur][i].l>p)
	{
		printf("%d %d\n",G[cur][i].l,G[cur][i].r);
		p=G[cur][i].r;
	}
}

G. Privatization of Roads in Treeland

题意：你可以选择 r 个不同的颜色给边涂色，如果一个点连接的所有边颜色不同，则为合法点，最多可以有k个非法点，要求你用最小的r去给树涂色。

思路：首先把所有点按照度数排序，那么第k+1个点的度数就是 r，加下来我们假设根为 1，然后bfs遍历树，假设我们也给点涂色，因为每个点至多只有一个父亲，设cnt为颜色，那么我们从cnt=color[fa]+1开始枚举颜色种类去给所有连接son的边以及son涂色。

#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=2e5+10;
int vis[maxn],ans[maxn],m,color[maxn];
vector<int>G[maxn],id[maxn];
struct node
{
	int u,du;
	bool operator<(const node&t)const
	{
		return du>t.du;
	}
}a[maxn];
void add(int u,int v,int i)
{
	G[u].push_back(v);
	id[u].push_back(i);
}
void bfs()
{
	queue<int>q;
	q.push(1);
	vis[1]=1;
	while(!q.empty())
	{
		int u=q.front();q.pop();
		int cnt=color[u];
		for(int i=0;i<G[u].size();i++)
		{
			int v=G[u][i];
			if(vis[v])continue;
			vis[v]=1;
			cnt++;
			if(cnt>m)cnt-=m;
			color[v]=cnt;
			int cur=id[u][i];
			ans[cur]=cnt;
			q.push(v);
		}
	}
}
int main()
{
	int n,k,u,v;
	scanf("%d%d",&n,&k);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v,i);
		add(v,u,i);	
		a[u].u=u;
		a[u].du++;
		a[v].u=v;
		a[v].du++;
	}
	sort(a+1,a+1+n);
	printf("%d\n",a[k+1].du);
	m=a[k+1].du;
	bfs();
	for(int i=1;i<n;i++)
	printf("%d ",ans[i]);
}