牛客月赛46

目录

A-赢的次数

B-子段和

C-英语作文

D-生活在树上

E-对决

F-数对

A-赢的次数

代码：

#include<iostream>
using namespace std;
int main()
{
	int n;
	cin>>n;
	if(n%2==0)cout<<n/2;
	else cout<<n/2<<' '<<n/2+1;
	return 0;
}

这题看完猜答案，没想到一试就对了，所以有些题目不要想太复杂

B-子段和

代码：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=5e5+10;
long long  a[N];
int n;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        if(!a[i])
        {
            cout<<"NO";
            return 0;
        }
    }
    sort(a+1,a+1+n);
    if(a[1]+a[n]==0)cout<<"NO";
    else cout<<"YES";
    return 0;
}

如果不知道结论，直接做，很容易超时，看了一些大佬的做法，才知道是有结论的。自己想不到其实可以通过样例来分析出所有情况，然后再列举出来。

C-英语作文

代码：

#include<iostream>
#include<string>
#include<map>
using namespace std;
typedef long long ll;
int main()
{
    int n,k;
    cin>>n>>k;
    map<string,int>mp;
    ll ans = 0;
    string s[n+1];
    for(int i=1;i<=n;i++) cin>>s[i];
    for(int i=1;i<=n;i++)
    {
        if(i-k-2>=1)  //如果数组越界
            mp[s[i-k-2]]--; //出了窗口就-- 
        ans += mp[s[i]];  //看看距离k的范围内有多少个配对的
        mp[s[i]]++;    //进窗口++
    }
    cout<<ans;
    return 0;
}

D-生活在树上

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e6+5;
vector<int> dis1[N],dis2[N];
int ans[N];
int main(){
	int n;
	cin>>n;
	int u,w;
	for(int i=2;i<=n;++i){
		cin>>u>>w;
		if(w==2){
			dis2[u].push_back(i);
			dis2[i].push_back(u);
		}
		else if(w==1){
			dis1[i].push_back(u);
			dis1[u].push_back(i);
		}
	}
	for(int i=1;i<=n;++i){
		ans[i]+=dis2[i].size();
		for(int j=0;j<dis1[i].size();++j){
			ans[i]+=dis1[dis1[i][j]].size();
		}
		printf("%d\n",ans[i]+1);  //这真的会卡endl
	}
	return 0;
}

E-对决
 

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1e6+10;
int n,k,a[N],s[N],b[N];
int main()
{
	cin>>n>>k;
	for (int i=1;i<=n;i++) cin>>a[i],b[i]=a[i];
	sort(b+1,b+1+n);
	for (int i=1;i<=n;i++)
	{
		int l,r;
		r=upper_bound(b+1,b+1+n,a[i])-b-1;
		l=k-r+1;
		if (l<=1&&(a[i]*2>=b[n]/2||a[i]*4+2>=b[n])) printf("1 ");
		else if (l==2&&r+1<n&&b[r+1]<=a[i]*2&&b[n]/2<=a[i]) printf("1 ");
		else printf("0 ");
	}
	return 0;
 } 

F-数对

#include<iostream>
typedef long long LL;
using namespace std;
LL T,n,k,ans;
LL q(LL x){
    return x*(x+1)/2;
}
int main(){
    cin>>T;
    while(T--){
        cin>>n>>k;
        if(k==1){
            cout<<q(n-1)<<endl;
            continue;
        }
        if(n>=k)ans=q(n-1)-q(k-2),n=k-1;
        else ans=0;
        if(k-1!=1&&1<=n&&k-1<=n)ans--;
        int tmp=(k+1)/2;
        if(tmp<=n)ans=ans-k*(n-tmp+1)+2*(q(n)-q(tmp-1));
        cout<<ans<<endl;
    }
    return 0;
}