题意:设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);
}
题意:选择尽可能多不相交的区间,使得这些区间的和相等。
思路: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]);
}