题解 | #英语作文#

牛客小白月赛46
题目链接：链接
英语作文
使用map中套vector来将数据离散化，然后对于每个字符串所对应的vector离散化数组进行操作。
对于vector中每个数据位置，使用二分来找到离他距离超过k的相同字符串的数量，并叠加起来，最后将总数输出

#define ll long long
#define mod 1000000007
#define eps 1e-7
#define endl "\n"
using namespace std;
const double PI = acos(-1);
const ll N = 2e5+ 10;
const ll maxn = 1e9+7;
void solve()
{
    int n,k;
    cin>>n>>k;
    string s;
    map<string,vector<int>> mp;
    for(int i=0;i<n;i++)
    {
        cin>>s;
        mp[s].push_back(i);
    }
    ll sum=0;
    for(auto i=mp.begin();i!=mp.end();i++)
    {
        vector<int>a(i->second);
        for(int j=0;j<a.size();j++)
        {
            sum+=upper_bound(a.begin(),a.end(),a[j]+k+1)-a.begin()-j-1;
        } 
    }
    cout<<sum<<endl;
}
int main()
{
    //init();
    int _ = 1;
    //scanf("%d", &_);
    while (_--)
    {
        solve();
    }
    return 0;
}