Neowise Labs Contest 1 (Codeforces Round 1018, Div. 1 + Div. 2)(A~E)

# 补题

A. Wonderful Sticks

题目:

思路:

从左到右,存一下该位置左边最大和最小,如果当前位置是'>'那么下一个位置就是mx+1,'<'就是mi-1,最后在处理一下让数组满足每一个数∈(1,n)。

代码:

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
ll inf=0x3f3f3f3f3f3f3f3f;
void solve(){
	int n;cin>>n;
	string s;cin>>s;
	vector<int> a(n);
	int  mx=100;
	int mi=100;
	a[0]=100;
	for(int i=0;i<s.size();i++){
		mi=min(mi,a[i]);
		mx=max(mx,a[i]);
		if(s[i]=='<')a[i+1]=mi-1;
		else a[i+1]=mx+1;
		if(i==s.size()-1){
			mi=min(mi,a[i+1]);
			mx=max(mx,a[i+1]);
		}
	}
	for(auto it:a)cout<<it-mi+1<<" ";cout<<endl;
	return;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int _;cin>>_;
	while(_--)solve();
	return 0;
}

B. Wonderful Gloves

题目:

思路:

考虑最坏的情况,先加上max(l[i],r[i]),然后对于min(l[i],r[i])组成的数组进行排序,然后取最大的k-1个,然后在+1就行了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
ll inf=0x3f3f3f3f3f3f3f3f;
void solve(){
	int n,k;cin>>n>>k;
	vector<int> l(n+1);
	vector<int> r(n+1);
	ll sm=0;
	for(int i=1;i<=n;i++)cin>>l[i];
	for(int i=1;i<=n;i++)cin>>r[i];
	vector<int> t;
	for(int i=1;i<=n;i++){
		sm+=max(l[i],r[i]);
		t.push_back(min(l[i],r[i]));
	}
	sort(t.begin(),t.end());
	ll t1=0;
	for(int i=n-1;i>n-1-k+1;i--){
		t1+=t[i];
	}
	cout<<sm+t1+1<<endl;
	return;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int _;cin>>_;
	while(_--)solve();
	return 0;
}

C. Wonderful City

题目:

思路:

首先可以发现它的的行和列是独立的,比如1 1,不管怎样选择a数组,这两个位置的数还是会相等。所以分别对行列使用一次dp就行了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
ll inf=0x3f3f3f3f3f3f3f3f;
void solve(){
	ll n;cin>>n;
	vector<vector<ll>> h(n+1,vector<ll> (n+1));
	for(ll i=1;i<=n;i++){
		for(ll j=1;j<=n;j++)cin>>h[i][j];
	}
	vector<ll> a(n+1);
	vector<ll> b(n+1);
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)cin>>b[i];
	vector<vector<ll>> dp1(n+1,vector<ll> (2,inf));
	ll res1=inf;
	dp1[1][1]=a[1];
	dp1[1][0]=0;
	for(ll i=1;i<n;i++){
		for(ll j1=0;j1<2;j1++){
			for(ll j2=0;j2<2;j2++){
				for(ll k=1;k<=n;k++){
					ll t1=h[i][k];
					ll t2=h[i+1][k];
					ll add=0;
					if(j1)t1++;
					if(j2)t2++,add+=a[i+1];
					if(t1==t2)break;
					if(k==n){
						dp1[i+1][j2]=min(dp1[i+1][j2],dp1[i][j1]+add);
					}
				}
			}
		}
	}
	res1=min(dp1[n][0],dp1[n][1]);
	ll res2=inf;
	vector<vector<ll>> dp2(n+1,vector<ll> (2,inf));
	dp2[1][0]=0;dp2[1][1]=b[1];
	for(ll i=1;i<n;i++){
		for(ll j1=0;j1<2;j1++){
			for(ll j2=0;j2<2;j2++){
				for(ll k=1;k<=n;k++){
					ll t1=h[k][i];
					ll t2=h[k][i+1];
					ll add=0;
					if(j1)t1++;
					if(j2)t2++,add+=b[i+1];
					if(t1==t2)break;
					if(k==n){
						dp2[i+1][j2]=min(dp2[i+1][j2],dp2[i][j1]+add);
					}
				}
			}
		}
	}
	res2=min(dp2[n][0],dp2[n][1]);
	if(res1==inf||res2==inf)cout<<-1<<endl;
	else cout<<res1+res2<<endl;
	return;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	ll _;cin>>_;
	while(_--)solve();
	return 0;
}

D. Wonderful Lightbulbs

题目:

思路:

通过题目可以观察到对一个点操作后的影响,可以发现假如图中的x,y是最开始的灯,那么无论后面怎样操作直线,L1,L2,上的灯的数量一定是奇数个,所以只需要求出满足条件的直线,然后他们的交点就是答案。

代码:

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
ll inf=0x3f3f3f3f3f3f3f3f;
void solve(){
	ll n;cin>>n;
	map<ll,ll> dx,dxy;
	for(ll i=0;i<n;i++){
		ll x,y;cin>>x>>y;
		dx[x]++;
		dxy[x+y]++;
	}
	ll ansx,ansxy;
	for(auto it:dx){
		if(it.second%2)ansx=it.first;
	}
	for(auto it:dxy){
		if(it.second%2)ansxy=it.first;
	}
	cout<<ansx<<" "<<ansxy-ansx<<endl;
	return;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	ll _;cin>>_;
	while(_--)solve();
	return 0;
}

E. Wonderful Teddy Bears

题目:

思路:

由题目我们知道我们每一次可以对三个连续的字符进行操作,然后我们可以可以发现,对于一个B,或者P它一次性走两步一定是比较好的,比如PBB->BBP,或者PPB->BPP,所以我们对于哪些BB或者PP我们可以优先移到左边或者右边,然后就会剩下PBPBPB......PB,此时如果有奇数个PB就拿去一个,其它的变成BBPPBBPP .....然后这里其实就是一个等差数列求和(PBPB->BBPP要两次操作别忘了加),最后再把PB里面的B移过去就行了。偶数就一样不过没有剩的PB。

代码:

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
typedef long long ll;
ll inf=0x3f3f3f3f3f3f3f3f;
void solve(){
	ll n;cin>>n;
	string s;cin>>s;
	s="#"+s;
	ll ans=0;
	ll cnt_p=0;
	ll t=0;
	stack<char> stk;
	for(ll i=1;i<=n;i++){
		if(s[i]=='B'){
			if(stk.size()){
				if(stk.top()=='B'){
					ans+=stk.size()/2;
					stk.pop();
				}else{
					stk.push('B');
				}
			}
			ans+=cnt_p;
		}else{
			if(stk.size()){
				if(stk.top()=='P'){
					cnt_p+=1;
					stk.pop();
				}else{
					stk.push('P');
				}
			}else{
				stk.push('P');
			}
		}
	}
	ll add=0;
	if(stk.size()){
		if(stk.top()=='P')stk.pop();
		ll t=stk.size()/2;
		if(t){
			if(t%2){
				t--;
				t/=2;
				add+=t*2+t*(t-1)+(stk.size()/2+1)/2;	
			}else{
				t/=2;
				add+=t*2+t*(t-1);
			}
		}
	}
	cout<<ans+add<<endl;
	return;
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	ll _;cin>>_;
	while(_--)solve();
	return 0;
}