为什么放在蓝桥平台可以AC?
查看原帖
为什么放在蓝桥平台可以AC?
1616821
__owen__楼主2025/1/20 22:21
#include <bits/stdc++.h>
using namespace std;
#define endl '\n';
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define int long long 
const int mod = 10007;
int ans;
void solve(){
	int n,m;cin>>n>>m;
	vector<int> nums(n + 1);
	vector<int> colors(n + 1);
	
	for(int i=1;i<=n;i++) cin>>nums[i];
	for(int i=1;i<=n;i++) cin>>colors[i];
	
	unordered_map<int,vector<int>> mp;
	
	for(int i=1;i<=n;i++){
		mp[colors[i]].push_back(i);
	}
	
	for(auto&c:mp){
		vector<int> v = c.second;
		int len = v.size();
		
		for(int i=0;i<len;i++){
			for(int j=i+1;j<len;j++){
				int x = v[i];
				int z = v[j];
				
				int y = (x + z) / 2;
				if(y > x && y < z && x % 2 == z % 2){
					ans = (ans + (x + z) * (nums[x] + nums[z])) % mod;
				}
				
			}
		}
	}
	cout << ans;
}
signed main() {
    IOS;
    int t = 1;
//    cin>>t;
    
    
    while(t--){
    	solve();
	}
    

    return 0;
}
2025/1/20 22:21
加载中...