Submission #628642

#	Time	Username	Problem	Language	Result	Execution time	Memory
628642		Iwanttobreakfree	Fancy Fence (CEOI20_fancyfence)	C++17	0 / 100	2 ms	340 KiB

fancyfence

#include <iostream>
#include <vector>
#include <map>
using namespace std;
const int mod=1e9+7;
int main(){
    int n;
    cin>>n;
    vector<int> v(n),h(n);
    map<int,int> mp;
    for(int& i:v)cin>>i;
    for(int& i:h)cin>>i;
    for(int i=0;i<n;i++){
        mp[h[i]]=(mp[h[i]]+v[i])%mod;
    }
    vector<pair<int,int>> comp;
    for(auto x:mp)comp.push_back(x);
    for(int i=comp.size()-1;i>0;i--)comp[i-1].second=(comp[i-1].second+comp[i].second)%mod;
    long long ans=0;
    for(int i=0;i<comp.size();i++){
        pair<int,int> x=comp[i];
        //cout<<(1ll*x.first*(x.first+1)/2)<<' '<<(1ll*x.second*(x.second+1)/2)<<'\n';
        ans+=(((1ll*x.first*(x.first+1)/2)%mod)*((1ll*x.second*(x.second+1)/2)%mod))%mod;
        if(i){
            pair<int,int> y=comp[i-1];
            ans-=(((1ll*y.first*(y.first+1)/2)%mod)*((1ll*x.second*(x.second+1)/2)%mod))%mod;
            if(ans<0)ans+=mod;
        }
    }
    cout<<ans;
}

Compilation message (stderr)

fancyfence.cpp: In function 'int main()':
fancyfence.cpp:20:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |     for(int i=0;i<comp.size();i++){
      |                 ~^~~~~~~~~~~~

• Subtask 1: 0 / 0
• Subtask 2: 0 / 12
• Subtask 3: 0 / 13
• Subtask 4: 0 / 15
• Subtask 5: 0 / 15
• Subtask 6: 0 / 18
• Subtask 7: 0 / 27